Add custom model name support

it would be cool to have custom model names as they could allow users to use new models faster.