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

added Expert to model to fix test
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Successful in 1m21s Details

This commit is contained in:
Falko Victor Habel 2025-03-28 16:10:22 +01:00
parent 5987a130f6
commit fb1b25507c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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()