RomanNum3ral
/
ai-toolkit
mirror of https://github.com/ostris/ai-toolkit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add ostris linear layer to linear layer searches.

This commit is contained in:
Jaret Burkett 2026-07-11 14:21:16 -06:00
parent 095d6e7418
commit 1d1e21177a
3 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
toolkit/lycoris_special.py
View File

@ -13,7 +13,8 @@ from toolkit.network_mixins import ToolkitNetworkMixin, ToolkitModuleMixin, Extr
# diffusers specific stuff
LINEAR_MODULES = [
'Linear',
'LoRACompatibleLinear'
'OstrisLinear',
'LoRACompatibleLinear',
]
CONV_MODULES = [
'Conv2d',

1
toolkit/memory_management/manager.py
View File

@ -6,6 +6,7 @@ LINEAR_MODULES = [
"Linear",
"LoRACompatibleLinear",
"QLinear",
'OstrisLinear',
]
CONV_MODULES = [
"Conv2d",

3
toolkit/models/DoRA.py
View File

@ -16,7 +16,8 @@ if TYPE_CHECKING:
# diffusers specific stuff
LINEAR_MODULES = [
'Linear',
'LoRACompatibleLinear'
'LoRACompatibleLinear',
'OstrisLinear',
# 'GroupNorm',
]
CONV_MODULES = [