wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-31 17:20:50 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Clean up

Browse Source
This commit is contained in:
Eric Lam 2025-05-20 21:55:28 -07:00
parent 59cc62c199
commit 309cc55004
1 changed files with 14 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
glob/manager_server.py
View File

@ -880,32 +880,19 @@ async def fetch_customnode_alternatives(request):
def check_model_installed(json_obj): def check_model_installed(json_obj):
def is_exists(model_dir_name, filename, url): def get_installed_model_paths(model_dir_name, filename, url):
if filename == '<huggingface>': if filename == '<huggingface>':
filename = os.path.basename(url) filename = os.path.basename(url)
dirs = folder_paths.get_folder_paths(model_dir_name) dirs = folder_paths.get_folder_paths(model_dir_name)
for x in dirs: full_paths = []
if os.path.exists(os.path.join(x, filename)):
return True
return False
def get_base_folder_path(model_dir_name, filename, url):
if filename == '<huggingface>':
filename = os.path.basename(url)
dirs = folder_paths.get_folder_paths(model_dir_name)
results = []
for x in dirs: for x in dirs:
full_path = os.path.join(x, filename) full_path = os.path.join(x, filename)
if os.path.exists(full_path): if os.path.exists(full_path):
results.append(full_path) full_paths.append(full_path)
return results return full_paths
model_dir_names = ['checkpoints', 'loras', 'vae', 'text_encoders', 'diffusion_models', 'clip_vision', 'embeddings', model_dir_names = ['checkpoints', 'loras', 'vae', 'text_encoders', 'diffusion_models', 'clip_vision', 'embeddings',
'diffusers', 'vae_approx', 'controlnet', 'gligen', 'upscale_models', 'hypernetworks', 'diffusers', 'vae_approx', 'controlnet', 'gligen', 'upscale_models', 'hypernetworks',
@ -925,10 +912,11 @@ def check_model_installed(json_obj):
if 'diffusion' not in item['filename'] and 'pytorch' not in item['filename'] and 'model' not in item['filename']: if 'diffusion' not in item['filename'] and 'pytorch' not in item['filename'] and 'model' not in item['filename']:
model_dir_name = model_dir_name_map.get(item['type'].lower()) model_dir_name = model_dir_name_map.get(item['type'].lower())
full_paths = get_base_folder_path(model_dir_name, item['filename'], item['url'])
# non-general name case # non-general name case
if item['filename'] in total_models_files: if item['filename'] in total_models_files:
full_paths = get_installed_model_paths(model_dir_name, item['filename'], item['url'])
item['installed'] = 'True' item['installed'] = 'True'
item['full_paths'] = full_paths item['full_paths'] = full_paths
return return
@ -936,14 +924,20 @@ def check_model_installed(json_obj):
if item['save_path'] == 'default': if item['save_path'] == 'default':
model_dir_name = model_dir_name_map.get(item['type'].lower()) model_dir_name = model_dir_name_map.get(item['type'].lower())
if model_dir_name is not None: if model_dir_name is not None:
item['installed'] = str(is_exists(model_dir_name, item['filename'], item['url'])) full_paths = get_installed_model_paths(model_dir_name, item['filename'], item['url'])
item['installed'] = str(len(fulls_paths) > 0)
item['full_paths'] = full_paths
else: else:
item['installed'] = 'False' item['installed'] = 'False'
else: else:
model_dir_name = item['save_path'].split('/')[0] model_dir_name = item['save_path'].split('/')[0]
if model_dir_name in folder_paths.folder_names_and_paths: if model_dir_name in folder_paths.folder_names_and_paths:
if is_exists(model_dir_name, item['filename'], item['url']): full_paths = get_installed_model_paths(model_dir_name, item['filename'], item['url'])
if len(full_paths) > 0:
item['installed'] = 'True' item['installed'] = 'True'
item['full_paths'] = full_paths
if 'installed' not in item: if 'installed' not in item:
item['installed'] = 'False' item['installed'] = 'False'