requirements.txt

absl-py
accelerate==0.30.1
annotated-types==0.7.0
bitsandbytes==0.42.0
certifi==2024.2.2
charset-normalizer==3.3.2
deepspeed==0.14.2
einops==0.8.0
filelock==3.14.0
flash-attn==2.6.1
fsspec==2024.5.0
grpcio
hjson==3.1.0
huggingface-hub==0.23.1
idna==3.7
importlib_metadata==7.1.0
Jinja2==3.1.4
joblib==1.4.2
MarkupSafe==2.1.5
mpmath==1.3.0
networkx==3.2.1
ninja==1.11.1.1
numpy==1.26.4
packaging==24.0
pandas==2.2.2
peft==0.11.1
pillow==10.3.0
protobuf==3.20.3
psutil==5.9.8
py-cpuinfo==9.0.0
pydantic==2.7.1
pydantic_core==2.18.2
pynvml==11.5.0
python-dateutil==2.9.0.post0
pytz==2024.1
PyYAML==6.0.1
regex==2024.5.15
requests==2.32.2
safetensors==0.4.3
scikit-learn==1.5.1
scipy==1.13.1
six
sympy==1.12
tensorboard==2.17.0
tensorboard-data-server==0.7.2
threadpoolctl==3.5.0
tiktoken==0.7.0
tokenizers==0.19.1
torch==2.3.0
torchaudio==2.3.0
torchvision==0.18.0
tqdm==4.66.4
transformers==4.41.1
transformers-stream-generator==0.0.4
triton==2.3.0
typing_extensions==4.12.0
tzdata==2024.1
urllib3==2.2.1
Werkzeug
zipp==3.18.2