alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 15:15:44 -03:00
Code Issues Packages Projects Releases Wiki Activity

feat: enhance folder path retrieval for diffusion models and improve warning messages

Browse Source
This commit is contained in:
Will Miao
2025-04-26 20:08:00 +08:00
parent 9e61bbbd8e
commit 2b5399c559
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
standalone.py
View File

@@ -13,15 +13,28 @@ class MockFolderPaths:
with open(settings_path, 'r', encoding='utf-8') as f: with open(settings_path, 'r', encoding='utf-8') as f:
settings = json.load(f) settings = json.load(f)
# Return paths if they exist in settings # For diffusion_models, combine unet and diffusers paths
if 'folder_paths' in settings and folder_name in settings['folder_paths']: if folder_name == "diffusion_models":
paths = settings['folder_paths'][folder_name] paths = []
if 'folder_paths' in settings:
if 'unet' in settings['folder_paths']:
paths.extend(settings['folder_paths']['unet'])
if 'diffusers' in settings['folder_paths']:
paths.extend(settings['folder_paths']['diffusers'])
# Filter out paths that don't exist # Filter out paths that don't exist
valid_paths = [p for p in paths if os.path.exists(p)] valid_paths = [p for p in paths if os.path.exists(p)]
if valid_paths: if valid_paths:
return valid_paths return valid_paths
else: else:
print(f"Warning: No valid paths found for {folder_name}") print(f"Warning: No valid paths found for {folder_name}")
# For other folder names, return their paths directly
elif 'folder_paths' in settings and folder_name in settings['folder_paths']:
paths = settings['folder_paths'][folder_name]
valid_paths = [p for p in paths if os.path.exists(p)]
if valid_paths:
return valid_paths
else:
print(f"Warning: No valid paths found for {folder_name}")
except Exception as e: except Exception as e:
print(f"Error loading folder paths from settings: {e}") print(f"Error loading folder paths from settings: {e}")