Spaces:
Sleeping
Sleeping
phoebehxf
commited on
Commit
·
784b9da
1
Parent(s):
eb2b25a
zerogpu
Browse files- requirements.txt +8 -5
requirements.txt
CHANGED
|
@@ -1,11 +1,11 @@
|
|
| 1 |
-
# PyTorch
|
| 2 |
-
torch==2.
|
| 3 |
-
torchvision==0.
|
| 4 |
-
torchaudio==2.
|
| 5 |
|
| 6 |
# Core dependencies
|
| 7 |
diffusers==0.29.0
|
| 8 |
-
transformers==4.41.0
|
| 9 |
accelerate==0.23.0
|
| 10 |
pyrallis
|
| 11 |
easydict
|
|
@@ -34,6 +34,9 @@ numpy==1.24.4
|
|
| 34 |
gradio
|
| 35 |
gradio-bbox-annotator
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
# Utilities
|
| 38 |
natsort
|
| 39 |
roifile
|
|
|
|
| 1 |
+
# PyTorch - ZeroGPU compatible versions
|
| 2 |
+
torch==2.1.0
|
| 3 |
+
torchvision==0.16.0
|
| 4 |
+
torchaudio==2.1.0
|
| 5 |
|
| 6 |
# Core dependencies
|
| 7 |
diffusers==0.29.0
|
| 8 |
+
transformers==4.41.0
|
| 9 |
accelerate==0.23.0
|
| 10 |
pyrallis
|
| 11 |
easydict
|
|
|
|
| 34 |
gradio
|
| 35 |
gradio-bbox-annotator
|
| 36 |
|
| 37 |
+
# ZeroGPU support
|
| 38 |
+
spaces
|
| 39 |
+
|
| 40 |
# Utilities
|
| 41 |
natsort
|
| 42 |
roifile
|