From 0653f59473d5afc499b887bd5363e094de06e030 Mon Sep 17 00:00:00 2001 From: Will Miao <13051207myq@gmail.com> Date: Thu, 3 Jul 2025 10:28:52 +0800 Subject: [PATCH] feat: Enhance relative path handling in download manager to include base model --- py/services/download_manager.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/py/services/download_manager.py b/py/services/download_manager.py index fc3b6448..ca11e297 100644 --- a/py/services/download_manager.py +++ b/py/services/download_manager.py @@ -96,10 +96,14 @@ class DownloadManager: return {'success': False, 'error': 'Default lora root path not set in settings'} save_dir = default_path - # Set relative_path to the first tag if available + # Set relative_path to version_info.baseModel/first_tag if available + base_model = version_info.get('baseModel', '') model_tags = version_info.get('model', {}).get('tags', []) - if model_tags: - relative_path = model_tags[0] + if base_model: + if model_tags: + relative_path = os.path.join(base_model, model_tags[0]) + else: + relative_path = base_model # Update save directory with relative path if provided if relative_path: