singhankit16 commited on
Commit
ae7a788
Β·
verified Β·
1 Parent(s): 87a3010

Update requirements.txt

Browse files

Adding torchvision in requirements

Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -5,4 +5,5 @@ torch>=2.1.0
5
  gradio>=4.0.0
6
  transformers @ git+https://github.com/huggingface/transformers.git
7
  Pillow>=10.0.0
8
- accelerate>=0.25.0
 
 
5
  gradio>=4.0.0
6
  transformers @ git+https://github.com/huggingface/transformers.git
7
  Pillow>=10.0.0
8
+ accelerate>=0.25.0
9
+ torchvision==0.25.0