Crocolil commited on
Commit
f672041
·
verified ·
1 Parent(s): ef4e083

Upload requirements.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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