updated config to match new version

2025-04-13 22:19:59 +02:00 · 2025-04-13 22:19:59 +02:00 · 1190f3c05f
parent 023ca07cf7
commit 1190f3c05f
1 changed files with 1 additions and 1 deletions
--- a/README.md
+++ b/README.md
@ -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