From 45f5cb46bd5905798676fa6718456fd4ba93b98f Mon Sep 17 00:00:00 2001
From: Will Miao <13051207myq@gmail.com>
Date: Wed, 10 Sep 2025 23:44:42 +0800
Subject: [PATCH] fix(utils): update base model retrieval to use model_data for
 consistency, fixes #423

---
 py/utils/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/py/utils/utils.py b/py/utils/utils.py
index 6ef973ad..2cf20a7c 100644
--- a/py/utils/utils.py
+++ b/py/utils/utils.py
@@ -154,7 +154,7 @@ def calculate_relative_path_for_model(model_data: Dict, model_type: str = 'lora'
 
     # For CivitAI models, prefer civitai data only if 'id' exists; for non-CivitAI models, use model_data directly
     if civitai_data and civitai_data.get('id') is not None:
-        base_model = civitai_data.get('baseModel', '')
+        base_model = model_data.get('base_model', '')
         # Get author from civitai creator data
         creator_info = civitai_data.get('creator') or {}
         author = creator_info.get('username') or 'Anonymous'