Skip to content

deps: limit dependency ranges #54

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
tedhtchang merged 9 commits into from
Jun 20, 2024
Merged

deps: limit dependency ranges #54

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
9b76df0
limit dependency ranges
anhuong Feb 23, 2024
e647586
merge changes from main
anhuong Feb 27, 2024
0ff6da3
merge changes from main branch
anhuong May 14, 2024
4359c7e
set upper limits for dependencies
anhuong May 14, 2024
c974a19
fix aim version, no need to pin to specific version
anhuong May 14, 2024
9e302ca
remove requirements file, no longer used
anhuong May 14, 2024
a05dda0
merge changes from main
anhuong Jun 20, 2024
6545ab1
deps below 1.0 have more constrained upper limit
anhuong Jun 20, 2024
33193d0
merge changes from main branch
anhuong Jun 20, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 14 additions & 14 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -26,24 +26,24 @@ classifiers=[
"Programming Language :: Python :: 3.11",
]
dependencies = [
"numpy",
"accelerate>=0.20.3",
"transformers",
"torch",
"sentencepiece",
"tokenizers>=0.13.3",
"tqdm",
"numpy>=1.26.4,<2.0",
"accelerate>=0.20.3,<0.40",
"transformers>=4.34.1,<5.0,!=4.38.2",
"torch>=2.2.0,<3.0",
"sentencepiece>=0.1.99,<0.3",
"tokenizers>=0.13.3,<1.0",
"tqdm>=4.66.2,<5.0",
"trl==0.8.6",
"peft>=0.8.0",
"datasets>=2.15.0",
"fire",
"simpleeval",
"peft>=0.8.0,<0.13",
"datasets>=2.15.0,<3.0",
"fire>=0.5.0,<1.0",
"simpleeval>=0.9.13,<1.0",
]

[project.optional-dependencies]
dev = ["wheel", "packaging", "ninja", "scikit-learn>=1.0, <2.0", "boto3"]
flash-attn = ["flash-attn"]
aim = ["aim==3.19.0"]
dev = ["wheel>=0.42.0,<1.0", "packaging>=23.2,<24", "ninja>=1.11.1.1,<2.0", "scikit-learn>=1.0, <2.0", "boto3>=1.34, <2.0"]
flash-attn = ["flash-attn>=2.5.3,<3.0"]
aim = ["aim>=3.19.0,<4.0"]
fms-accel = [
"fms_acceleration @ git+https://github.com/foundation-model-stack/fms-acceleration.git#subdirectory=plugins/framework"
]
Expand Down
15 changes: 0 additions & 15 deletions requirements.txt
View file
Open in desktop

This file was deleted.