Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
c271014
1
Parent(s):
8fc3c7f
fixing error on not reference issue
Browse files- model/pre_download.py +6 -3
model/pre_download.py
CHANGED
|
@@ -17,8 +17,9 @@ def pre_download_image_models():
|
|
| 17 |
import imagen_hub
|
| 18 |
errored_models = []
|
| 19 |
for model_string in IMAGE_GENERATION_MODELS:
|
| 20 |
-
|
| 21 |
model_lib, model_name, model_type = model_string.split("_")
|
|
|
|
| 22 |
|
| 23 |
if model_lib == "imagenhub":
|
| 24 |
try:
|
|
@@ -39,8 +40,9 @@ def pre_download_image_models():
|
|
| 39 |
import imagen_hub
|
| 40 |
errored_models = []
|
| 41 |
for model_string in IMAGE_EDITION_MODELS:
|
| 42 |
-
|
| 43 |
model_lib, model_name, model_type = model_string.split("_")
|
|
|
|
| 44 |
|
| 45 |
if model_lib == "imagenhub":
|
| 46 |
try:
|
|
@@ -61,8 +63,9 @@ def pre_download_video_models():
|
|
| 61 |
import videogen_hub
|
| 62 |
errored_models = []
|
| 63 |
for model_string in VIDEO_GENERATION_MODELS:
|
| 64 |
-
|
| 65 |
model_lib, model_name, model_type = model_string.split("_")
|
|
|
|
| 66 |
|
| 67 |
if model_lib == "videogenhub":
|
| 68 |
try:
|
|
|
|
| 17 |
import imagen_hub
|
| 18 |
errored_models = []
|
| 19 |
for model_string in IMAGE_GENERATION_MODELS:
|
| 20 |
+
|
| 21 |
model_lib, model_name, model_type = model_string.split("_")
|
| 22 |
+
print("Loading image generation model:", model_name)
|
| 23 |
|
| 24 |
if model_lib == "imagenhub":
|
| 25 |
try:
|
|
|
|
| 40 |
import imagen_hub
|
| 41 |
errored_models = []
|
| 42 |
for model_string in IMAGE_EDITION_MODELS:
|
| 43 |
+
|
| 44 |
model_lib, model_name, model_type = model_string.split("_")
|
| 45 |
+
print("Loading image edition model:", model_name)
|
| 46 |
|
| 47 |
if model_lib == "imagenhub":
|
| 48 |
try:
|
|
|
|
| 63 |
import videogen_hub
|
| 64 |
errored_models = []
|
| 65 |
for model_string in VIDEO_GENERATION_MODELS:
|
| 66 |
+
|
| 67 |
model_lib, model_name, model_type = model_string.split("_")
|
| 68 |
+
print("Loading video generation model:", model_name)
|
| 69 |
|
| 70 |
if model_lib == "videogenhub":
|
| 71 |
try:
|