SUPERSHANKY commited on
Commit
e3737e4
1 Parent(s): 4ccfec7

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -15,7 +15,7 @@ prettytable==3.6.0
15
  pytorch-lightning
16
  safetensors==0.3.0
17
  timm
18
- https://download.pytorch.org/whl/cu118/torch-1.13.1%2Bcu117-cp39-cp39-linux_x86_64.whl
19
  torchvision==0.14.1+cu117
20
  fastai
21
  transformers==4.27.1
@@ -28,4 +28,5 @@ Triton==2.0.0
28
  git+https://github.com/huggingface/diffusers.git
29
  scipy
30
  ftfy
31
- psutil
 
 
15
  pytorch-lightning
16
  safetensors==0.3.0
17
  timm
18
+ https://download.pytorch.org/whl/cu117/torch-1.13.1%2Bcu117-cp39-cp39-linux_x86_64.whl
19
  torchvision==0.14.1+cu117
20
  fastai
21
  transformers==4.27.1
 
28
  git+https://github.com/huggingface/diffusers.git
29
  scipy
30
  ftfy
31
+ psutil
32
+ Xformers==0.0.17.dev476