ne.from_pretrained(
...     "runwayml/stable-diffusion-v1-5", vae=vae, torch_dtype=torch.float16
... ).to("cuda")

>>> pipe("horse", generator=torch.manual_seed(0)).images  wrapper  < source > ( *args **kwargs  )     disable_slicing  < source > ( )    Disable sliced VAE decoding. If enable_slicing was previously enabled, this method will go back to computing
decoding in one step.  disable_tiling  < source > ( )    Disable tiled VAE decoding. If