From 81599b8f439bb773579e6ff913458b887dfb0ace Mon Sep 17 00:00:00 2001 From: Will Miao <13051207myq@gmail.com> Date: Fri, 14 Mar 2025 11:22:21 +0800 Subject: [PATCH] Fix: correct LoRA naming issue when fetching data from Civitai --- py/routes/api_routes.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/py/routes/api_routes.py b/py/routes/api_routes.py index de147e78..07d3471a 100644 --- a/py/routes/api_routes.py +++ b/py/routes/api_routes.py @@ -354,7 +354,7 @@ class ApiRoutes: # Update model name if available if 'model' in civitai_metadata: if civitai_metadata.get('model', {}).get('name'): - local_metadata['model_name'] = determine_base_model(civitai_metadata['model']['name']) + local_metadata['model_name'] = civitai_metadata['model']['name'] # Fetch additional model metadata (description and tags) if we have model ID model_id = civitai_metadata['modelId'] @@ -365,7 +365,7 @@ class ApiRoutes: local_metadata['tags'] = model_metadata.get('tags', []) # Update base model - local_metadata['base_model'] = civitai_metadata.get('baseModel') + local_metadata['base_model'] = determine_base_model(civitai_metadata.get('baseModel')) # Update preview if needed if not local_metadata.get('preview_url') or not os.path.exists(local_metadata['preview_url']):