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
|