Upload folder using huggingface_hub
Browse files- 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 |
-
|
|
|
|
| 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)
|