From b20ba1f27cbd4e1c84cf8ec72b345723de9e7c80 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Thu, 28 Aug 2025 00:45:02 +0800
Subject: [PATCH] Fix #9537 (#9576)

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

diff --git a/comfy/weight_adapter/lokr.py b/comfy/weight_adapter/lokr.py
index 49b0be55f..563c835f5 100644
--- a/comfy/weight_adapter/lokr.py
+++ b/comfy/weight_adapter/lokr.py
@@ -97,6 +97,9 @@ class LoKrAdapter(WeightAdapterBase):
             (mat1, mat2, alpha, None, None, None, None, None, None)
         )
 
+    def to_train(self):
+        return LokrDiff(self.weights)
+
     @classmethod
     def load(
         cls,