|
|
|
@ -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
|
|
|
|
|