added support for getting raw model