alphard/ComfyUI-Lora-Auto-Trigger-Words
1
0
Fork 0
mirror of https://github.com/idrirap/ComfyUI-Lora-Auto-Trigger-Words.git synced 2026-03-21 21:22:12 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix: tags were never pulled from civitai the first time they were requested

Browse Source
This commit is contained in:
Dijkstra
2023-11-20 18:11:23 +01:00
parent 8af8cce210
commit 8ed3d36eaf
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
utils.py
View File

@@ -103,21 +103,21 @@ def load_and_save_tags(lora_name, force_fetch):
output_tags_list = [] output_tags_list = []
lora_path = folder_paths.get_full_path("loras", lora_name) lora_path = folder_paths.get_full_path("loras", lora_name)
if lora_tags is None or force_fetch: # search on civitai only if no local cache or forced if lora_tags is None or force_fetch or output_tags is None: # search on civitai only if no local cache or forced
print("calculating lora hash") print("[Lora-Auto-Trigger] calculating lora hash")
LORAsha256 = calculate_sha256(lora_path) LORAsha256 = calculate_sha256(lora_path)
print("requesting infos") print("[Lora-Auto-Trigger] requesting infos")
model_info = get_model_version_info(LORAsha256) model_info = get_model_version_info(LORAsha256)
if model_info is not None: if model_info is not None:
if "trainedWords" in model_info: if "trainedWords" in model_info:
print("tags found!") print("[Lora-Auto-Trigger] tags found!")
if lora_tags is None: if lora_tags is None:
lora_tags = {} lora_tags = {}
lora_tags[lora_name] = model_info["trainedWords"] lora_tags[lora_name] = model_info["trainedWords"]
save_dict_to_json(lora_tags,json_tags_path) save_dict_to_json(lora_tags, json_tags_path)
output_tags_list = model_info["trainedWords"] output_tags_list = model_info["trainedWords"]
else: else:
print("No informations found.") print("[Lora-Auto-Trigger] No informations found.")
if lora_tags is None: if lora_tags is None:
lora_tags = {} lora_tags = {}
lora_tags[lora_name] = [] lora_tags[lora_name] = []