Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

fixed spelling error

This commit is contained in:
Falko Victor Habel 2025-01-26 13:10:24 +01:00
parent e5a5618160
commit 59b2784e92
2 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/aiia/model/Model.py
View File

@ -222,9 +222,4 @@ class AIIArecursive(AIIA):
processed_patches.append(pp)
combined_output = torch.mean(torch.stack(processed_patches, dim=0), dim=0)
return combined_output
config = AIIAConfig()
model2 = AIIABaseShared(config)
model2.save("shared")
return combined_output

2
src/aiia/model/__init__.py
View File

@ -1,2 +1,2 @@
from .config import AIIAConfig
from .Model import AIIA, AIIABase, AIIAchunked, AIIAExpert, AIIAmoe, AIIAresursive
from .Model import AIIA, AIIABase, AIIAchunked, AIIAExpert, AIIAmoe, AIIArecursive