diff --git a/pyproject.toml b/pyproject.toml index 03fd1178..11b9916e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -42,7 +42,7 @@ dependencies = [ [project.optional-dependencies] dev = ["wheel", "packaging", "ninja", "scikit-learn>=1.0, <2.0"] flash-attn = ["flash-attn"] -aim = ["aim==3.18.1"] +aim = ["aim==3.19.0"] [tool.setuptools.packages.find] exclude = ["tests", "tests.*"] diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 00000000..4d5c0378 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,15 @@ +numpy +accelerate>=0.20.3 +packaging +transformers>=4.34.1 +torch +aim==3.19.0 +sentencepiece +tokenizers>=0.13.3 +tqdm +trl +ninja +peft>=0.8.0 +datasets>=2.15.0 +fire +