from aiia import AIIABase

model = AIIABase.load(path="AIIA-base-512", precision="bf16")