Support loading different archs
from hear21passt.base import get_basic_model
model = get_basic_model(mode="logits", arch="passt_s_kd_p16_128_ap486")
from hear21passt.base import get_basic_model
model = get_basic_model(mode="logits", arch="passt_s_kd_p16_128_ap486")