Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
0ddac36
1
Parent(s):
d43fe7c
debug
Browse files- model/model_manager.py +2 -2
- requirements.txt +2 -2
model/model_manager.py
CHANGED
@@ -81,10 +81,10 @@ class ModelManager:
|
|
81 |
self.enable_nsfw = enable_nsfw
|
82 |
self.load_guard(enable_nsfw)
|
83 |
self.loaded_models = {}
|
84 |
-
if do_pre_download:
|
85 |
-
pre_download_all_models(include_video=False)
|
86 |
if do_debug_packages:
|
87 |
debug_packages()
|
|
|
|
|
88 |
|
89 |
try:
|
90 |
self.generate_image_ig.zerogpu = True
|
|
|
81 |
self.enable_nsfw = enable_nsfw
|
82 |
self.load_guard(enable_nsfw)
|
83 |
self.loaded_models = {}
|
|
|
|
|
84 |
if do_debug_packages:
|
85 |
debug_packages()
|
86 |
+
if do_pre_download:
|
87 |
+
pre_download_all_models(include_video=False)
|
88 |
|
89 |
try:
|
90 |
self.generate_image_ig.zerogpu = True
|
requirements.txt
CHANGED
@@ -7,7 +7,7 @@ h5py
|
|
7 |
xformers
|
8 |
numpy>=1.23.5
|
9 |
pandas<2.0.0
|
10 |
-
peft
|
11 |
torch
|
12 |
torchvision
|
13 |
torchaudio
|
@@ -25,7 +25,7 @@ pillow~=9.5.0
|
|
25 |
einops>=0.4.1
|
26 |
torch-fidelity>=0.3.0
|
27 |
setuptools>=59.5.0
|
28 |
-
transformers
|
29 |
torchmetrics>=0.6.0
|
30 |
kornia>=0.6
|
31 |
diffusers>=0.18.0
|
|
|
7 |
xformers
|
8 |
numpy>=1.23.5
|
9 |
pandas<2.0.0
|
10 |
+
peft~=0.10.0
|
11 |
torch
|
12 |
torchvision
|
13 |
torchaudio
|
|
|
25 |
einops>=0.4.1
|
26 |
torch-fidelity>=0.3.0
|
27 |
setuptools>=59.5.0
|
28 |
+
transformers~=4.37.2
|
29 |
torchmetrics>=0.6.0
|
30 |
kornia>=0.6
|
31 |
diffusers>=0.18.0
|