vinesmsuic commited on
Commit
b3b25b3
1 Parent(s): 4a07334

catch error

Browse files
Files changed (1) hide show
  1. 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