Fix load some model in layout

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