Update requirements.txt

Add torch version limit
2024-05-31 11:19:14 +08:00
parent 6e75e236b6
commit 9cae83066e
1 changed files with 2 additions and 2 deletions
@@ -4,7 +4,7 @@ safetensors
 timm
 tomesd
 accelerate
-torch
+torch>=2.2.0
 torchdiffeq
 torchsde
 xformers
@@ -21,4 +21,4 @@ imageio[pyav]
 tensorboard
 gradio==3.41.2
 diffusers==0.27.0
-transformers==4.37.2
+transformers==4.37.2