updated config to match new version
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Has been cancelled Details

This commit is contained in:
Falko Victor Habel 2025-04-13 22:19:59 +02:00
parent 023ca07cf7
commit 1190f3c05f
1 changed files with 1 additions and 1 deletions

View File

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