fix: update model type checks to include LoCon and lycoris in API routes. Fixes https://github.com/willmiao/ComfyUI-Lora-Manager/issues/159

This commit is contained in:
Will Miao
2025-05-05 07:48:08 +08:00
parent a3d8261686
commit 1ff2019dde
3 changed files with 8 additions and 13 deletions

View File

@@ -426,10 +426,11 @@ class StandardMetadataParser(RecipeMetadataParser):
resources = json.loads(resources_json)
# Filter loras and checkpoints
for resource in resources:
if resource.get('type') == 'lora':
# 确保 weight 字段被正确保留
# Process both 'lora' and 'lycoris' types as loras
if resource.get('type') == 'lora' or resource.get('type') == 'lycoris':
# Ensure weight field is properly preserved
lora_entry = resource.copy()
# 如果找不到 weight默认为 1.0
# Default to 1.0 if weight not found
if 'weight' not in lora_entry:
lora_entry['weight'] = 1.0
# Ensure modelVersionName is included