Added experimental orbit quant

2026-07-09 15:14:51 -06:00
parent a4bbe167ce
commit 5a9b5bde3f
6 changed files with 770 additions and 7 deletions
--- a/toolkit/network_mixins.py
+++ b/toolkit/network_mixins.py
@@ -29,7 +29,8 @@ Module = Union['LoConSpecialModule', 'LoRAModule', 'DoRAModule']
 LINEAR_MODULES = [
    'Linear',
    'LoRACompatibleLinear',
-    'QLinear'
+    'QLinear',
+    'OstrisLinear',
    # 'GroupNorm',
 ]
 CONV_MODULES = [