alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-24 06:32:12 -03:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Simplify model download handling by consolidating download logic and updating parameter usage

Browse Source
This commit is contained in:
Will Miao
2025-07-02 18:25:42 +08:00
parent 9d8b7344cd
commit d7cb546c5f
7 changed files with 25 additions and 186 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
py/utils/routes_common.py
View File

@@ -587,15 +587,14 @@ class ModelRouteUtils:
})
# Check which identifier is provided
download_url = data.get('download_url')
model_hash = data.get('model_hash')
model_id = data.get('model_id')
model_version_id = data.get('model_version_id')
# Validate that at least one identifier is provided
if not any([download_url, model_hash, model_version_id]):
# Only model_id is required, model_version_id is optional
if not model_id:
return web.Response(
status=400,
text="Missing required parameter: Please provide either 'download_url', 'hash', or 'modelVersionId'"
text="Missing required parameter: Please provide 'model_id'"
)
# Use the correct root directory based on model type
@@ -603,8 +602,7 @@ class ModelRouteUtils:
save_dir = data.get(root_key)
result = await download_manager.download_from_civitai(
download_url=download_url,
model_hash=model_hash,
model_id=model_id,
model_version_id=model_version_id,
save_dir=save_dir,
relative_path=data.get('relative_path', ''),