nodes/lilly1987-ComfyUI_node_Lilly
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix 230711

Browse Source 
LoraLoaderTextRandom is not test
This commit is contained in:
lilly
2023-07-11 22:27:58 +09:00
parent ab31903931
commit ed401cb9b8
3 changed files with 59 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
LoraLoaderText.py
+28 -4
View File
@@ -11,6 +11,9 @@ else:
from .mypath import *
class LoraLoaderText:
def __init__(self):
self.loaded_lora = None
@classmethod
def INPUT_TYPES(s):
return {
@@ -34,16 +37,37 @@ class LoraLoaderText:
def load_lora(self, model, clip, lora_name, strength_model, strength_clip):
print(f"[{ccolor}]lora_name : [/{ccolor}]", lora_name)
if strength_model == 0 and strength_clip == 0:
print("[red]strength_model,strength_clip 0[/red] : ", lora_name)
return (model, clip)
if lora_name is None or lora_name =="" :
print("[red]No lora_name[/red] : ", lora_name)
return (model, clip)
lora_path=getFullPath(lora_name,"loras")
lora_path = folder_paths.get_full_path("loras", lora_name)
if lora_path is None:
print("[red]No lora_path of lora_name [/red] : ", lora_name)
return (model, clip)
#print("[red]No lora_path of lora_name [/red] : ", lora_name)
lora_path=getFullPath(lora_name,"loras")
if lora_path is None:
print("[red]No lora_path of lora_name [/red] : ", lora_name)
return (model, clip)
lora = None
if self.loaded_lora is not None:
if self.loaded_lora[0] == lora_path:
lora = self.loaded_lora[1]
else:
del self.loaded_lora
if lora is None:
lora = comfy.utils.load_torch_file(lora_path, safe_load=True)
self.loaded_lora = (lora_path, lora)
# =========================================
try:
model_lora, clip_lora = comfy.sd.load_lora_for_models(model, clip, lora_path, strength_model, strength_clip)
model_lora, clip_lora = comfy.sd.load_lora_for_models(model, clip, lora, strength_model, strength_clip)
return (model_lora, clip_lora)
except Exception as e:
console.print_exception()