phoebehxf commited on
Commit
8b7cac3
·
1 Parent(s): 5914a8a
Files changed (1) hide show
  1. requirements.txt +7 -7
requirements.txt CHANGED
@@ -2,34 +2,34 @@
2
  torch==2.4.1
3
  torchvision==0.19.1
4
  torchaudio==2.4.1
 
5
  # Core dependencies
6
  diffusers==0.29.0
7
- # transformers==4.37.2
8
  transformers==4.40.0
9
- huggingface_hub==0.24.1
10
  accelerate==0.23.0
11
  pyrallis
12
  easydict
13
  omegaconf==2.1.1
14
  einops==0.3.0
15
  torch-fidelity==0.3.0
16
- torchmetrics>=0.11.0
17
- pytorch-lightning==2.0.9
18
  taming-transformers @ git+https://github.com/CompVis/taming-transformers.git@master
19
- clip @ git+https://github.com/openai/CLIP.git@main
20
 
21
  # Computer vision
22
  opencv-python==4.5.5.64
23
  opencv-python-headless==4.5.5.64
24
  kornia==0.7.1
25
  albumentations==0.4.3
26
- imageio>=2.27 # ✅ 保证兼容 scikit-image 0.21.0
27
  imageio-ffmpeg==0.4.2
28
  matplotlib
29
  scikit-image==0.21.0
30
  Pillow
31
  segment-anything
32
- numpy==1.24.4 # ✅ 避免 numpy 2.x 引发 ABI 问题
33
 
34
  # Gradio
35
  gradio
 
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.40.0
9
+ huggingface_hub>=1.0.0 # Let Spaces use its version
10
  accelerate==0.23.0
11
  pyrallis
12
  easydict
13
  omegaconf==2.1.1
14
  einops==0.3.0
15
  torch-fidelity==0.3.0
16
+ torchmetrics==1.3.0 # ✅ UPDATE THIS
17
+ pytorch-lightning==2.2.0 # ✅ UPDATE THIS (compatible with newer torchmetrics)
18
  taming-transformers @ git+https://github.com/CompVis/taming-transformers.git@master
19
+ clip @ git+https://github.com/OpenAI/CLIP.git@main
20
 
21
  # Computer vision
22
  opencv-python==4.5.5.64
23
  opencv-python-headless==4.5.5.64
24
  kornia==0.7.1
25
  albumentations==0.4.3
26
+ imageio>=2.27
27
  imageio-ffmpeg==0.4.2
28
  matplotlib
29
  scikit-image==0.21.0
30
  Pillow
31
  segment-anything
32
+ numpy==1.24.4
33
 
34
  # Gradio
35
  gradio