Ketengan-Diffusion-Lab commited on
Commit
0e18231
·
verified ·
1 Parent(s): 7d197cd

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -2
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  wheel
2
  setuptools
3
  transformers
4
- --extra-index-url https://download.pytorch.org/whl/cu118
5
- torch==1.10.0+cu118 torchvision==0.11.0+cu118
6
  gradio
7
  accelerate
8
  flash_attn
 
1
  wheel
2
  setuptools
3
  transformers
4
+ torch==1.10.0
5
+ torchvision==0.11.0
6
  gradio
7
  accelerate
8
  flash_attn