K446 commited on
Commit
00b8117
·
1 Parent(s): d4e5470

Add torchvision and hf_transfer

Browse files
Files changed (1) hide show
  1. requirements-training.txt +3 -1
requirements-training.txt CHANGED
@@ -1,10 +1,12 @@
1
- # Training deps (torch, torchao, 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
 
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