Spaces:
Running
on
Zero
Running
on
Zero
Update requirements.txt
Browse files- requirements.txt +4 -1
requirements.txt
CHANGED
|
@@ -2,7 +2,10 @@
|
|
| 2 |
git+https://github.com/huggingface/transformers.git
|
| 3 |
peft>=0.11.0
|
| 4 |
accelerate
|
| 5 |
-
torch>=2.1.0
|
| 6 |
gradio>=4.33
|
| 7 |
huggingface_hub>=0.30.0,<1.0
|
| 8 |
pillow
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
git+https://github.com/huggingface/transformers.git
|
| 3 |
peft>=0.11.0
|
| 4 |
accelerate
|
|
|
|
| 5 |
gradio>=4.33
|
| 6 |
huggingface_hub>=0.30.0,<1.0
|
| 7 |
pillow
|
| 8 |
+
|
| 9 |
+
# --- core ---
|
| 10 |
+
torch==2.1.2 # pin to avoid pip pulling 2.7.x
|
| 11 |
+
torchvision==0.16.2 # matching wheel for 2.1.x
|