zxdu20's picture youwei's picture
add missing `model_type` (#9)
05b0e4e