diff --git a/toolkit/lycoris_special.py b/toolkit/lycoris_special.py
index 84021b49..be2250b2 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 0785f06c..ddf03450 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 653575e9..3cfeb4a2 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 = [