JoelWester commited on
Commit
756e1a8
·
verified ·
1 Parent(s): c960e53

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +11 -3
requirements.txt CHANGED
@@ -1,17 +1,25 @@
1
- torch==2.1
 
2
  torchvision==0.16.0
3
- huggingface_hub<0.26.0
 
 
 
 
 
4
  diffusers==0.25.1
5
  transformers==4.46.1
6
  accelerate==1.0.1
7
  datasets==3.1.0
 
 
8
  xformers==0.0.22.post7
9
  bitsandbytes==0.44.1
10
  face-alignment==1.4.1
11
  albumentations==1.4.1
12
 
 
13
  safetensors
14
  Pillow
15
- numpy
16
  opencv-python
17
  gradio
 
1
+ # Core versions matching the original environment.yml
2
+ torch==2.1.0
3
  torchvision==0.16.0
4
+
5
+ # Fix for NumPy ABI compatibility with torch
6
+ numpy==1.26.4
7
+
8
+ # HF stack pinned to what this repo expects
9
+ huggingface_hub==0.25.2
10
  diffusers==0.25.1
11
  transformers==4.46.1
12
  accelerate==1.0.1
13
  datasets==3.1.0
14
+
15
+ # Extra libs from the original env
16
  xformers==0.0.22.post7
17
  bitsandbytes==0.44.1
18
  face-alignment==1.4.1
19
  albumentations==1.4.1
20
 
21
+ # General utilities + UI
22
  safetensors
23
  Pillow
 
24
  opencv-python
25
  gradio