fixes from_model function and adds tests (#921)
* fixes from_model function and adds tests
* removes mock accelerator from tests
* fixes tests by adding model_configs where needed
* Apply suggestion from @Copilot
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
* fixes from review
* fixes from review
* fixes from review
* fixes from review
---------
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
