Update snake_oil.py

This commit is contained in:
Doctor Diffusion
2024-12-30 00:58:46 -05:00
committed by GitHub
parent 6de078ba3d
commit 76cb09a7f8
+2 -2
View File
@@ -33,10 +33,10 @@ class NegativeLoRALoader:
}
RETURN_TYPES = ("MODEL",)
FUNCTION = "apply_inverted_LoRA"
FUNCTION = "apply_inverted_lora" # Corrected the method name here
CATEGORY = "🐍🛢️ Snake Oil"
def apply_snake_oil(self, model, nlora, inverted_strength_value):
def apply_inverted_lora(self, model, nlora, inverted_strength_value): # Corrected the method name here
if inverted_strength_value == 0:
print("inverted_strength_value is 0, returning the original model.")
return (model,)