updated config to match new version
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Has been cancelled
Details
Gitea Actions For AIIA / Explore-Gitea-Actions (push) Has been cancelled
Details
This commit is contained in:
parent
023ca07cf7
commit
1190f3c05f
|
@ -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
|
||||||
|
|
Loading…
Reference in New Issue