feat/tf_support #37

Merged
Fabel merged 13 commits from feat/tf_support into develop 2025-04-16 20:59:48 +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 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