chore: minor optim changes (add apollo, improve docs, remove lion-pytorch) (#2444)

* feat: add apollo-torch

* chore: update optimizer list

* fix: deleted accidental requirements file

* fix: remove mention of deprecated lion_pytorch
This commit is contained in:
NanoCode012
2025-03-27 05:14:07 +07:00
committed by GitHub
parent 2c34a4634e
commit e2da821e67
3 changed files with 30 additions and 324 deletions

View File

@@ -16,9 +16,7 @@ def parse_requirements():
with open("./requirements.txt", encoding="utf-8") as requirements_file:
lines = [r.strip() for r in requirements_file.readlines()]
for line in lines:
is_extras = (
"deepspeed" in line or "mamba-ssm" in line or "lion-pytorch" in line
)
is_extras = "deepspeed" in line or "mamba-ssm" in line
if line.startswith("--extra-index-url"):
# Handle custom index URLs
_, url = line.split()
@@ -135,15 +133,15 @@ setup(
"mlflow": [
"mlflow",
],
"lion-pytorch": [
"lion-pytorch==0.1.2",
],
"galore": [
"galore_torch",
],
"apollo": [
"apollo-torch",
],
"optimizers": [
"galore_torch",
"lion-pytorch==0.1.2",
"apollo-torch",
"lomo-optim==0.1.1",
"torch-optimi==0.2.1",
],