Spaces:
Running
Running
Upload requirements.txt with huggingface_hub
Browse files- requirements.txt +19 -0
requirements.txt
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# App runtime
|
| 2 |
+
gradio==6.18.0
|
| 3 |
+
pandas==3.0.3
|
| 4 |
+
requests==2.34.2
|
| 5 |
+
Pillow==12.2.0
|
| 6 |
+
|
| 7 |
+
# ML — recognition & recommendation models (inference + training scripts)
|
| 8 |
+
datasets==5.0.0
|
| 9 |
+
transformers==5.12.0
|
| 10 |
+
accelerate==1.14.0
|
| 11 |
+
peft==0.19.1
|
| 12 |
+
trl==1.6.0
|
| 13 |
+
|
| 14 |
+
# torch: left unpinned on purpose — Colab/HF Spaces ship a CUDA-matched build,
|
| 15 |
+
# and pinning here would force a CPU reinstall over it. Tested with 2.12.0.
|
| 16 |
+
torch>=2.12
|
| 17 |
+
|
| 18 |
+
# Optional, only needed for `train_recommender.py --load-in-4bit`
|
| 19 |
+
# bitsandbytes==0.49.0
|