RioShiina commited on
Commit
f1bf026
·
verified ·
1 Parent(s): e62785a

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. utils/app_utils.py +3 -2
utils/app_utils.py CHANGED
@@ -322,7 +322,8 @@ def _ensure_model_downloaded(display_name: str, progress=gr.Progress()):
322
  if display_name not in ALL_MODEL_MAP:
323
  raise ValueError(f"Model '{display_name}' not found in configuration.")
324
 
325
- _, repo_filename, _, _ = ALL_MODEL_MAP[display_name]
 
326
  base_filename = os.path.basename(repo_filename)
327
 
328
  download_info = ALL_FILE_DOWNLOAD_MAP.get(base_filename)
@@ -502,7 +503,7 @@ def ensure_ipadapter_models_downloaded(preset_name: str, progress):
502
  temp_display_name = f"ipadapter_asset_{filename}"
503
 
504
  if temp_display_name not in ALL_MODEL_MAP:
505
- ALL_MODEL_MAP[temp_display_name] = (None, filename, model_type, None)
506
 
507
  try:
508
  _ensure_model_downloaded(temp_display_name, progress)
 
322
  if display_name not in ALL_MODEL_MAP:
323
  raise ValueError(f"Model '{display_name}' not found in configuration.")
324
 
325
+ model_info = ALL_MODEL_MAP[display_name]
326
+ repo_filename = model_info[1]
327
  base_filename = os.path.basename(repo_filename)
328
 
329
  download_info = ALL_FILE_DOWNLOAD_MAP.get(base_filename)
 
503
  temp_display_name = f"ipadapter_asset_{filename}"
504
 
505
  if temp_display_name not in ALL_MODEL_MAP:
506
+ ALL_MODEL_MAP[temp_display_name] = (None, filename, model_type, None, None)
507
 
508
  try:
509
  _ensure_model_downloaded(temp_display_name, progress)