Clean up model loading.