zhiweili
commited on
Commit
·
3c7f45d
1
Parent(s):
fc1393b
change torch version
Browse files- requirements.txt +2 -2
requirements.txt
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
gradio
|
| 2 |
-
torch==2.
|
| 3 |
-
torch_tensorrt==2.
|
| 4 |
torchvision==0.20.0
|
| 5 |
diffusers
|
| 6 |
transformers
|
|
|
|
| 1 |
gradio
|
| 2 |
+
torch==2.4.0
|
| 3 |
+
torch_tensorrt==2.4.0
|
| 4 |
torchvision==0.20.0
|
| 5 |
diffusers
|
| 6 |
transformers
|