osanseviero cbensimon HF staff commited on
Commit
57c304a
1 Parent(s): 53924ae

CUDA error fix attempt (#3)

Browse files

- CUDA error fix attempt (cf786ab343c638d382491e8f4fc9f8aa1885bbf4)


Co-authored-by: Charles Bensimon <cbensimon@users.noreply.huggingface.co>

Files changed (1) hide show
  1. requirements.txt +4 -2
requirements.txt CHANGED
@@ -1,9 +1,11 @@
 
 
 
 
1
  fairscale
2
  ftfy
3
  Pillow
4
  timm
5
- torch
6
- torchvision
7
  transformers==4.15.0
8
  -e git+https://github.com/openai/CLIP.git@main#egg=clip
9
  -e git+https://github.com/pharmapsychotic/BLIP.git@main#egg=blip
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu113
2
+ torch
3
+ torchvision
4
+
5
  fairscale
6
  ftfy
7
  Pillow
8
  timm
 
 
9
  transformers==4.15.0
10
  -e git+https://github.com/openai/CLIP.git@main#egg=clip
11
  -e git+https://github.com/pharmapsychotic/BLIP.git@main#egg=blip