develop #41

Merged
Fabel merged 27 commits from develop into main 2025-04-17 17:08:57 +00:00
1 changed files with 1 additions and 1 deletions
Showing only changes of commit 1190f3c05f - Show all commits

View File

@ -29,7 +29,7 @@ from aiia.model import AIIAConfig
from aiia.pretrain import Pretrainer
# Create your model
config = AIIAConfig(model_name="AIIA-Base-512x20k")
config = AIIAConfig(model_type="AIIA-Base-512x20k")
model = AIIABase(config)
# Initialize pretrainer with the model