youwei's picture
add missing `model_type`
d51ed93