Add torchvision and hf_transfer
Browse files
requirements-training.txt
CHANGED
|
@@ -1,10 +1,12 @@
|
|
| 1 |
-
# Training deps (torch,
|
| 2 |
transformers>=4.51.3,<5.0
|
| 3 |
trl>=0.12.0,<1.0
|
| 4 |
peft>=0.13.0
|
| 5 |
accelerate>=1.0.0
|
| 6 |
datasets>=3.0.0
|
| 7 |
bitsandbytes
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# Shared with environment
|
| 10 |
fastapi
|
|
|
|
| 1 |
+
# Training deps (torch, unsloth installed separately in Dockerfile)
|
| 2 |
transformers>=4.51.3,<5.0
|
| 3 |
trl>=0.12.0,<1.0
|
| 4 |
peft>=0.13.0
|
| 5 |
accelerate>=1.0.0
|
| 6 |
datasets>=3.0.0
|
| 7 |
bitsandbytes
|
| 8 |
+
torchvision
|
| 9 |
+
hf_transfer
|
| 10 |
|
| 11 |
# Shared with environment
|
| 12 |
fastapi
|