From 1d1e21177a8c6a22a4f95f9ef842162ad21363ed Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Sat, 11 Jul 2026 14:21:16 -0600
Subject: [PATCH] Add ostris linear layer to linear layer searches.

---
 toolkit/lycoris_special.py           | 3 ++-
 toolkit/memory_management/manager.py | 1 +
 toolkit/models/DoRA.py               | 3 ++-
 3 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/toolkit/lycoris_special.py b/toolkit/lycoris_special.py
index 84021b4..be2250b 100644
--- a/toolkit/lycoris_special.py
+++ b/toolkit/lycoris_special.py
@@ -13,7 +13,8 @@ from toolkit.network_mixins import ToolkitNetworkMixin, ToolkitModuleMixin, Extr
 # diffusers specific stuff
 LINEAR_MODULES = [
     'Linear',
-    'LoRACompatibleLinear'
+    'OstrisLinear',
+    'LoRACompatibleLinear',
 ]
 CONV_MODULES = [
     'Conv2d',
diff --git a/toolkit/memory_management/manager.py b/toolkit/memory_management/manager.py
index 0785f06..ddf0345 100644
--- a/toolkit/memory_management/manager.py
+++ b/toolkit/memory_management/manager.py
@@ -6,6 +6,7 @@ LINEAR_MODULES = [
     "Linear",
     "LoRACompatibleLinear",
     "QLinear",
+    'OstrisLinear',
 ]
 CONV_MODULES = [
     "Conv2d",
diff --git a/toolkit/models/DoRA.py b/toolkit/models/DoRA.py
index 653575e..3cfeb4a 100644
--- a/toolkit/models/DoRA.py
+++ b/toolkit/models/DoRA.py
@@ -16,7 +16,8 @@ if TYPE_CHECKING:
 # diffusers specific stuff
 LINEAR_MODULES = [
     'Linear',
-    'LoRACompatibleLinear'
+    'LoRACompatibleLinear',
+    'OstrisLinear',
     # 'GroupNorm',
 ]
 CONV_MODULES = [