fix 230711

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
+1
View File
@@ -1,2 +1,3 @@
*.pyc *.pyc
**/nppBackup/
+28 -4
View File
@@ -11,6 +11,9 @@ else:
from .mypath import * from .mypath import *
class LoraLoaderText: class LoraLoaderText:
def __init__(self):
self.loaded_lora = None
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
@@ -34,16 +37,37 @@ class LoraLoaderText:
def load_lora(self, model, clip, lora_name, strength_model, strength_clip): def load_lora(self, model, clip, lora_name, strength_model, strength_clip):
print(f"[{ccolor}]lora_name : [/{ccolor}]", lora_name) 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 =="" : if lora_name is None or lora_name =="" :
print("[red]No lora_name[/red] : ", lora_name) print("[red]No lora_name[/red] : ", lora_name)
return (model, clip) return (model, clip)
lora_path=getFullPath(lora_name,"loras") lora_path = folder_paths.get_full_path("loras", lora_name)
if lora_path is None: if lora_path is None:
print("[red]No lora_path of lora_name [/red] : ", lora_name) #print("[red]No lora_path of lora_name [/red] : ", lora_name)
return (model, clip) 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: 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) return (model_lora, clip_lora)
except Exception as e: except Exception as e:
console.print_exception() console.print_exception()
+30 -7
View File
@@ -48,18 +48,41 @@ class LoraLoaderTextRandom:
strength_model=random.uniform(min(strength_model_min,strength_model_max),max(strength_model_min,strength_model_max)) strength_model=random.uniform(min(strength_model_min,strength_model_max),max(strength_model_min,strength_model_max))
strength_clip=random.uniform(min(strength_clip_min,strength_clip_max),max(strength_clip_min,strength_clip_max)) strength_clip=random.uniform(min(strength_clip_min,strength_clip_max),max(strength_clip_min,strength_clip_max))
lora_path =check_name_pt("loras",name_split_choice(lora_name)) print(f"[{ccolor}]lora_name : [/{ccolor}]", lora_name)
if lora_path is None: if strength_model == 0 and strength_clip == 0:
print(f"LoraLoaderTextRandom : {lora_name} is none") print("[red]strength_model,strength_clip 0[/red] : ", lora_name)
return (model, clip) return (model, clip)
print(f"LoraLoaderTextRandom : {lora_path} ")
if lora_name is None or lora_name =="" :
print("[red]No lora_name[/red] : ", lora_name)
return (model, clip)
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)
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: 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_path, strength_model, strength_clip)
return (model_lora, clip_lora) return (model_lora, clip_lora)
except Exception as e: except Exception as e:
print(f"LoraLoaderTextRandom Exception : {e}" ) console.print_exception()
err_msg = traceback.format_exc()
print(err_msg)
return (model, clip) return (model, clip)
#NODE_CLASS_MAPPINGS = { #NODE_CLASS_MAPPINGS = {