Add ostris linear layer to linear layer searches.

2026-07-11 14:21:16 -06:00
parent 095d6e7418
commit 1d1e21177a
3 changed files with 5 additions and 2 deletions
--- 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 = [