evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add ostris linear layer to linear layer searches.

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-07-11 14:21:16 -06:00
1 parent 095d6e7418
commit 1d1e21177a
3 files changed
+5 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/memory_management/manager.py
+1
View File
@@ -6,6 +6,7 @@ LINEAR_MODULES = [
"Linear",
"LoRACompatibleLinear",
"QLinear",
'OstrisLinear',
]
CONV_MODULES = [
"Conv2d",