Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
Commit
•
b3b25b3
1
Parent(s):
4a07334
catch error
Browse files- model/pre_download.py +2 -0
model/pre_download.py
CHANGED
@@ -8,6 +8,8 @@ def pre_download_all_models(include_video=True):
|
|
8 |
imagedit_dl_error = pre_download_image_models_edit()
|
9 |
if include_video:
|
10 |
videogen_dl_error = pre_download_video_models_gen()
|
|
|
|
|
11 |
print("All models downloaded.")
|
12 |
print("Models that encountered download error:", "Image Generation:", imagen_dl_error, "Image Edition:", imagedit_dl_error, "Video Generation:", videogen_dl_error)
|
13 |
|
|
|
8 |
imagedit_dl_error = pre_download_image_models_edit()
|
9 |
if include_video:
|
10 |
videogen_dl_error = pre_download_video_models_gen()
|
11 |
+
else:
|
12 |
+
videogen_dl_error = ["None"]
|
13 |
print("All models downloaded.")
|
14 |
print("Models that encountered download error:", "Image Generation:", imagen_dl_error, "Image Edition:", imagedit_dl_error, "Video Generation:", videogen_dl_error)
|
15 |
|