ddriscoll commited on
Commit
a5e1544
·
verified ·
1 Parent(s): 2c967f4

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +9 -2
requirements.txt CHANGED
@@ -6,13 +6,20 @@ requests==2.31.0
6
  wikipedia==1.4.0
7
 
8
  # PyTorch
 
9
  torch==2.0.1
10
 
11
  # Hugging Face Ecosystem
12
  transformers==4.31.0
13
- huggingface_hub==0.15.1 # Updated version
 
 
 
 
14
  diffusers==0.19.3
15
- accelerate==0.20.3 # optional but often needed for device_map="auto"
 
 
16
 
17
  # Real-ESRGAN and Dependencies
18
  realesrgan==0.3.0
 
6
  wikipedia==1.4.0
7
 
8
  # PyTorch
9
+ # If GPU is required, replace with the correct CUDA/MPS version
10
  torch==2.0.1
11
 
12
  # Hugging Face Ecosystem
13
  transformers==4.31.0
14
+
15
+ # IMPORTANT: pin huggingface_hub to 0.14.x so that cached_download is available
16
+ huggingface_hub==0.14.1
17
+
18
+ # diffusers 0.19.x references cached_download, so do NOT upgrade beyond 0.19.x
19
  diffusers==0.19.3
20
+
21
+ # accelerate is optional but often required for device_map="auto"
22
+ accelerate==0.20.3
23
 
24
  # Real-ESRGAN and Dependencies
25
  realesrgan==0.3.0