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

feat/smoe #34

Merged
Fabel merged 7 commits from feat/smoe into main 2025-03-28 15:48:40 +00:00
Conversation 0 Commits 7 Files Changed 7 +1 -1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit fb1b25507c - Show all commits

2
tests/model/test_aiia.py
View File

@ -1,6 +1,6 @@
import os
import torch
from aiia import AIIABase, AIIABaseShared, AIIAExpert, AIIAmoe, AIIAchunked, AIIAConfig, AIIASparseMoe
from aiia.model.Model import AIIABase, AIIABaseShared, AIIAExpert, AIIAmoe, AIIAchunked, AIIAConfig, AIIASparseMoe
def test_aiiabase_creation():
config = AIIAConfig()