diff --git a/comfy/weight_adapter/lora.py b/comfy/weight_adapter/lora.py index 60145d5b1..729dbd9e6 100644 --- a/comfy/weight_adapter/lora.py +++ b/comfy/weight_adapter/lora.py @@ -74,6 +74,9 @@ class LoRAAdapter(WeightAdapterBase): (mat1, mat2, alpha, None, None, None) ) + def to_train(self): + return LoraDiff(self.weights) + @classmethod def load( cls,