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

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
- torch==1.10.0
5
- torchvision==0.11.0
6
  gradio
7
  accelerate
8
  flash_attn
 
1
  wheel
2
  setuptools
3
  transformers
4
+ torch
5
+ torchvision
6
  gradio
7
  accelerate
8
  flash_attn