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/memory_management/manager.py
+++ b/toolkit/memory_management/manager.py
@@ -6,6 +6,7 @@ LINEAR_MODULES = [
    "Linear",
    "LoRACompatibleLinear",
    "QLinear",
+    'OstrisLinear',
 ]
 CONV_MODULES = [
    "Conv2d",