ford442 commited on
Commit
00bf447
·
verified ·
1 Parent(s): 725bdfd

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,8 +1,10 @@
1
  torch
 
 
2
  git+https://github.com/huggingface/accelerate.git
3
  #bitsandbytes
4
  git+https://github.com/huggingface/transformers.git
5
- git+https://github.com/huggingface/optimum.git
6
  #git+https://github.com/facebookresearch/xformers.git
7
  hf-transfer
8
  huggingface_hub[hf_transfer]
 
1
  torch
2
+ torchvision
3
+ torchaudio
4
  git+https://github.com/huggingface/accelerate.git
5
  #bitsandbytes
6
  git+https://github.com/huggingface/transformers.git
7
+ #git+https://github.com/huggingface/optimum.git
8
  #git+https://github.com/facebookresearch/xformers.git
9
  hf-transfer
10
  huggingface_hub[hf_transfer]