mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 14:42:11 -03:00
feat(misc): mark model versions in library for Civitai user models
This commit is contained in:
@@ -687,9 +687,6 @@ class ModelLibraryHandler:
|
||||
except (TypeError, ValueError):
|
||||
continue
|
||||
|
||||
if await scanner.check_model_version_exists(version_id_int):
|
||||
continue
|
||||
|
||||
images = version.get("images") or []
|
||||
thumbnail_url = None
|
||||
if images and isinstance(images, list):
|
||||
@@ -697,6 +694,8 @@ class ModelLibraryHandler:
|
||||
if isinstance(first_image, dict):
|
||||
thumbnail_url = first_image.get("url")
|
||||
|
||||
in_library = await scanner.check_model_version_exists(version_id_int)
|
||||
|
||||
versions.append(
|
||||
{
|
||||
"modelId": model_id_int,
|
||||
@@ -707,6 +706,7 @@ class ModelLibraryHandler:
|
||||
"tags": tags,
|
||||
"baseModel": version.get("baseModel"),
|
||||
"thumbnailUrl": thumbnail_url,
|
||||
"inLibrary": in_library,
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user