Corrected the model's config name so that it works properly (#2)

commit from chengluli