phoebehxf commited on
Commit
784b9da
·
1 Parent(s): eb2b25a
Files changed (1) hide show
  1. requirements.txt +8 -5
requirements.txt CHANGED
@@ -1,11 +1,11 @@
1
- # PyTorch 2.4.1 + torchvision
2
- torch==2.4.1
3
- torchvision==0.19.1
4
- torchaudio==2.4.1
5
 
6
  # Core dependencies
7
  diffusers==0.29.0
8
- transformers==4.41.0 # ✅ This version supports huggingface-hub 1.x
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