|
|
|
@ -572,6 +572,19 @@ class LoraLoader:
|
|
|
|
|
model_lora, clip_lora = comfy.sd.load_lora_for_models(model, clip, lora, strength_model, strength_clip)
|
|
|
|
|
return (model_lora, clip_lora)
|
|
|
|
|
|
|
|
|
|
class LoraLoaderModelOnly(LoraLoader):
|
|
|
|
|
@classmethod
|
|
|
|
|
def INPUT_TYPES(s):
|
|
|
|
|
return {"required": { "model": ("MODEL",),
|
|
|
|
|
"lora_name": (folder_paths.get_filename_list("loras"), ),
|
|
|
|
|
"strength_model": ("FLOAT", {"default": 1.0, "min": -20.0, "max": 20.0, "step": 0.01}),
|
|
|
|
|
}}
|
|
|
|
|
RETURN_TYPES = ("MODEL",)
|
|
|
|
|
FUNCTION = "load_lora_model_only"
|
|
|
|
|
|
|
|
|
|
def load_lora_model_only(self, model, lora_name, strength_model):
|
|
|
|
|
return (self.load_lora(model, None, lora_name, strength_model, 0)[0],)
|
|
|
|
|
|
|
|
|
|
class VAELoader:
|
|
|
|
|
@staticmethod
|
|
|
|
|
def vae_list():
|
|
|
|
@ -1703,6 +1716,7 @@ NODE_CLASS_MAPPINGS = {
|
|
|
|
|
|
|
|
|
|
"ConditioningZeroOut": ConditioningZeroOut,
|
|
|
|
|
"ConditioningSetTimestepRange": ConditioningSetTimestepRange,
|
|
|
|
|
"LoraLoaderModelOnly": LoraLoaderModelOnly,
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
|
|
|