Using IGPUModel instead of pickles for loading.