image_encoder.eval()
```  
You'll also load the [`WuerstchenPrior`] model for optimization.  
```py
prior = WuerstchenPrior.from_pretrained(args.pretrained_prior_model_name_or_path, subfolder="prior")