From 17ff8196cb12ff49aa536792ed0fec149c9f0626 Mon Sep 17 00:00:00 2001
From: Sergey Borisov <stalkek7779@yandex.ru>
Date: Tue, 6 Aug 2024 04:32:24 +0300
Subject: [PATCH] Remove tmp code

---
 invokeai/backend/lora.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/invokeai/backend/lora.py b/invokeai/backend/lora.py
index 098b0fe0d7..b0366cc6ce 100644
--- a/invokeai/backend/lora.py
+++ b/invokeai/backend/lora.py
@@ -218,8 +218,8 @@ class LoKRLayer(LoRALayerBase):
 
         self.w1 = values.get("lokr_w1", None)
         if self.w1 is None:
-            self.w1_a = values.get("lokr_w1_a", None)
-            self.w1_b = values.get("lokr_w1_b", None)
+            self.w1_a = values["lokr_w1_a"]
+            self.w1_b = values["lokr_w1_b"]
         else:
             self.w1_b = None
             self.w1_a = None