alphard/Bjornulf_custom_nodes
1
0
Fork 0
mirror of https://github.com/justUmen/Bjornulf_custom_nodes.git synced 2026-03-21 20:52:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

0.77

Browse Source
This commit is contained in:
justumen
2025-03-19 17:36:25 +01:00
parent 44d69e8907
commit 39dfb0220a
76 changed files with 3207 additions and 955 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
loader_lora_with_path.py
View File

@@ -8,42 +8,43 @@ class LoaderLoraWithPath:
return {
"required": {
"model": ("MODEL",),
"clip": ("CLIP",),
"lora_path": ("STRING", {"default": ""}),
"strength_model": ("FLOAT", {"default": 1.0, "min": -100.0, "max": 100.0, "step": 0.01}),
"strength_clip": ("FLOAT", {"default": 1.0, "min": -100.0, "max": 100.0, "step": 0.01}),
},
"optional": {
"clip": ("CLIP",),
}
}
RETURN_TYPES = ("MODEL", "CLIP")
FUNCTION = "load_lora" # Added this line
RETURN_TYPES = ("MODEL", "CLIP", "STRING")
FUNCTION = "load_lora"
CATEGORY = "Bjornulf"
def load_lora(self, model, clip, lora_path, strength_model, strength_clip):
def load_lora(self, model, lora_path, strength_model, strength_clip, clip=None):
try:
# Check if path exists
if not os.path.isfile(lora_path):
print(f"Error: Lora file not found at path: {lora_path}")
return (model, clip)
return (model, clip if clip is not None else None, lora_path if lora_path is not None else None)
# Load the Lora file
try:
lora = comfy.utils.load_torch_file(lora_path)
except Exception as e:
print(f"Error loading Lora file: {str(e)}")
return (model, clip)
lora = comfy.utils.load_torch_file(lora_path)
if clip is not None:
model_lora, clip_lora = comfy.sd.load_lora_for_models(
model, clip, lora, strength_model, strength_clip
)
return (model_lora, clip_lora, lora_path if lora_path is not None else None)
else:
model_lora = model.clone()
# Assuming ModelPatcher with diffusion_model
state_dict = model_lora.model.diffusion_model.state_dict()
for key in lora:
if 'unet' in key: # Filter for UNet keys; adjust as needed
if key in state_dict:
state_dict[key] += strength_model * lora[key]
model_lora.model.diffusion_model.load_state_dict(state_dict)
return (model_lora, None, lora_path if lora_path is not None else None)
# Apply the Lora
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:
print(f"Error in load_lora: {str(e)}")
return (model, clip)
return (model, clip if clip is not None else None)