From 1190f3c05ff08c9c0192e25051c7b1549ca86fc6 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Sun, 13 Apr 2025 22:19:59 +0200
Subject: [PATCH] updated config to match new version

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 333838b..90e2a0b 100644
--- 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