Spaces:
Sleeping
Sleeping
add torchvision
Browse files- requirements.txt +1 -0
requirements.txt
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
torch==2.1.1 # if you encounter CUDA verision mismatches error, try to install torch manually to specify the version that matches your cuda version
|
|
|
|
| 2 |
xformers==0.0.23
|
| 3 |
accelerate==0.24.1
|
| 4 |
diffusers==0.19.0
|
|
|
|
| 1 |
torch==2.1.1 # if you encounter CUDA verision mismatches error, try to install torch manually to specify the version that matches your cuda version
|
| 2 |
+
torchvision==0.16.1
|
| 3 |
xformers==0.0.23
|
| 4 |
accelerate==0.24.1
|
| 5 |
diffusers==0.19.0
|