nodes/myshell-ai-ComfyUI-ShellAgent-Plugin
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

handle model_searcher fail

Browse Source
This commit is contained in:
wl-zhao
2024-10-23 10:46:48 +08:00
parent fc1f9afbcd
commit 00fb0b91cd
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dependency_checker.py
+10 -6
View File
@@ -100,7 +100,10 @@ def fetch_model_searcher_results(model_ids):
} }
response = requests.post(url, headers=headers, json=data) response = requests.post(url, headers=headers, json=data)
results = [item[:10] for item in response.json()] if response.status_code == 200:
results = [item[:10] for item in response.json()]
else:
results = None
return results return results
def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes and models at the same time def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes and models at the same time
@@ -195,11 +198,12 @@ def resolve_dependencies(prompt, custom_dependencies): # resolve custom nodes an
# try to fetch from myshell model searcher # try to fetch from myshell model searcher
missing_model_results_myshell = fetch_model_searcher_results(missing_model_ids) missing_model_results_myshell = fetch_model_searcher_results(missing_model_ids)
for missing_model_id, missing_model_urls in zip(missing_model_ids, missing_model_results_myshell): if missing_model_results_myshell is not None:
if len(missing_model_urls) > 0: for missing_model_id, missing_model_urls in zip(missing_model_ids, missing_model_results_myshell):
models_dict[missing_model_id]["require_recheck"] = False if len(missing_model_urls) > 0:
models_dict[missing_model_id]["urls"] = missing_model_urls models_dict[missing_model_id]["require_recheck"] = False
print("successfully fetch results from myshell", models_dict[missing_model_id]) models_dict[missing_model_id]["urls"] = missing_model_urls
print("successfully fetch results from myshell", models_dict[missing_model_id])
# step 3: handle local files # step 3: handle local files
process_local_file_path_async(file_mapping_dict, max_workers=20) process_local_file_path_async(file_mapping_dict, max_workers=20)