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 13:12:13 -03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #13 from idrirap/issue12

Browse Source 
fix: broken preview when using both enable preview and override_lora_name
This commit is contained in:
dijkstra
2024-03-14 23:26:17 +01:00
committed by GitHub
parent d5641ba745 f3815b7ce4
commit 9d8b79844f
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
nodes_autotrigger.py
View File

@@ -129,7 +129,9 @@ class LoraLoaderAdvanced:
def load_lora(self, model, clip, lora_name, strength_model, strength_clip, force_fetch, enable_preview, append_loraname_if_empty, override_lora_name=""):
if override_lora_name != "":
lora_name = {"content": override_lora_name, "image": ""}
has_preview, prev = get_preview_path(override_lora_name, "loras")
prev = f"loras/{prev}" if has_preview else None
lora_name = {"content": override_lora_name, "image": prev, "type": "loras"}
meta_tags_list = sort_tags_by_frequency(get_metadata(lora_name["content"], "loras"))
civitai_tags_list = load_and_save_tags(lora_name["content"], force_fetch)
@@ -192,7 +194,9 @@ class LoraLoaderStackedAdvanced:
def set_stack(self, lora_name, lora_weight, force_fetch, enable_preview, append_loraname_if_empty, lora_stack=None, override_lora_name=""):
if override_lora_name != "":
lora_name = {"content": override_lora_name, "image": ""}
has_preview, prev = get_preview_path(override_lora_name, "loras")
prev = f"loras/{prev}" if has_preview else None
lora_name = {"content": override_lora_name, "image": prev, "type": "loras"}
civitai_tags_list = load_and_save_tags(lora_name["content"], force_fetch)

2
utils.py
View File

@@ -31,6 +31,8 @@ def copy_preview_to_temp(file_name):
lora_less = "/".join(file_name.split("/")[1:])
file_path = folder_paths.get_full_path("loras", lora_less)
if file_path is None:
return None, None
temp_path = folder_paths.get_temp_directory()
preview_path = os.path.join(temp_path, "lora_preview")