vinesmsuic commited on
Commit
c271014
1 Parent(s): 8fc3c7f

fixing error on not reference issue

Browse files
Files changed (1) hide show
  1. 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
- print("Loading image generation model:", model_name)
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
- print("Loading image edition model:", model_name)
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
- print("Loading video generation model:", model_name)
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: