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

fix(model-library): update response structure to return model versions directly

Browse Source
This commit is contained in:
Will Miao
2025-10-04 22:06:33 +08:00
parent 837138db49
commit 9d22dd3465
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
py/routes/handlers/misc_handlers.py
View File

@@ -474,18 +474,18 @@ class ModelLibraryHandler:
embedding_versions = await embedding_scanner.get_model_versions_by_id(model_id) embedding_versions = await embedding_scanner.get_model_versions_by_id(model_id)
model_type = None model_type = None
version_ids: list[int] = [] versions = []
if lora_versions: if lora_versions:
model_type = "lora" model_type = "lora"
version_ids = [version["versionId"] for version in lora_versions] versions = lora_versions
elif checkpoint_versions: elif checkpoint_versions:
model_type = "checkpoint" model_type = "checkpoint"
version_ids = [version["versionId"] for version in checkpoint_versions] versions = checkpoint_versions
elif embedding_versions: elif embedding_versions:
model_type = "embedding" model_type = "embedding"
version_ids = [version["versionId"] for version in embedding_versions] versions = embedding_versions
return web.json_response({"success": True, "modelType": model_type, "modelVersionIds": version_ids}) return web.json_response({"success": True, "modelType": model_type, "versions": versions})
except Exception as exc: # pragma: no cover - defensive logging except Exception as exc: # pragma: no cover - defensive logging
logger.error("Failed to check model existence: %s", exc, exc_info=True) logger.error("Failed to check model existence: %s", exc, exc_info=True)
return web.json_response({"success": False, "error": str(exc)}, status=500) return web.json_response({"success": False, "error": str(exc)}, status=500)