Poprawki do ładowania modeli

Signed-off-by: Grzes <kabza.grzegorz@gmail.com>