wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2026-02-15 23:52:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update manager_server.py

Browse Source 
the download_model_base variable replaces the folder_paths.models_dir value in fullpath
This commit is contained in:
autoscancode 2025-04-02 19:55:43 +08:00 committed by GitHub
parent 908a1009d2
commit ba635711fb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
glob/manager_server.py
View File

@ -272,13 +272,12 @@ import json
import zipfile import zipfile
import urllib.request import urllib.request
if 'download_model_base' in folder_paths.folder_names_and_paths:
models_base = folder_paths.folder_names_and_paths['download_model_base'][0][0]
logging.info(f"models_base download_model_base: {models_base}")
else:
models_base = folder_paths.models_dir
def get_model_dir(data, show_log=False): def get_model_dir(data, show_log=False):
if 'download_model_base' in folder_paths.folder_names_and_paths:
models_base = folder_paths.folder_names_and_paths['download_model_base'][0][0]
else:
models_base = folder_paths.models_dir
# NOTE: Validate to prevent path traversal. # NOTE: Validate to prevent path traversal.
if any(char in data['filename'] for char in {'/', '\\', ':'}): if any(char in data['filename'] for char in {'/', '\\', ':'}):
return None return None
@ -926,7 +925,7 @@ def check_model_installed(json_obj):
else: else:
filename = item['filename'] filename = item['filename']
fullpath = os.path.join(folder_paths.models_dir, item['save_path'], filename) fullpath = os.path.join(models_base, item['save_path'], filename)
item['installed'] = 'True' if os.path.exists(fullpath) else 'False' item['installed'] = 'True' if os.path.exists(fullpath) else 'False'