Use the `weight_name` parameter to specify the specific weight file and the `prefix` parameter to filter for the appropriate state dicts (`"unet"` in this case) to load.  
```py
from diffusers import AutoPipelineForText2Image
import torch