modify model

Signed-off-by: xiaokangfrost <xiaokangfrost@gmail.com>