requirements.txt

albumentations==1.4.7
codecarbon==2.3.5
datasets[vision]~=2.19.0
evaluate==0.4.1
ipadic==1.0.0
jiwer==3.0.3
joblib==1.4.0
loguru==0.7.2
pandas==2.2.2
nltk==3.8.1
optuna==3.6.1
Pillow==10.3.0
protobuf==4.23.4
sacremoses==0.1.1
scikit-learn==1.5.0
sentencepiece==0.2.0
tqdm==4.66.4
werkzeug==3.0.2
xgboost==2.0.3
huggingface_hub==0.23.4
requests==2.31.0
einops==0.7.0
invisible-watermark==0.2.0
packaging==24.0
cryptography==42.0.5
nvitop==1.3.2
# latest versions
tensorboard==2.16.2
peft==0.11.1
trl==0.9.4
tiktoken==0.6.0
transformers==4.40.2
optimum==1.21.2
numpy==1.26.4
accelerate==0.27.2
diffusers==0.27.2
bitsandbytes==0.43.1
# extras
rouge_score==0.1.2
py7zr==0.21.0
fastapi==0.111.0
uvicorn==0.29.0
python-multipart==0.0.9
pydantic==2.7.1
hf-transfer
pyngrok==7.1.6
authlib==1.3.0
itsdangerous==2.2.0
seqeval==1.2.2
httpx==0.27.0
pyyaml==6.0.1
timm==1.0.7
torchmetrics==1.4.0
pycocotools==2.0.7
sentence-transformers==3.0.1
optimum-habana==1.12.0