From b8757c5822c696ca16b523c36aa3681d7cf23bcf Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Tue, 10 Jun 2025 23:20:30 +0800
Subject: [PATCH] Update lora.py

---
 comfy/weight_adapter/lora.py | 3 +++
 1 file changed, 3 insertions(+)

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,