Spaces:
Running
on
Zero
Running
on
Zero
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 |
|