Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py

Co-authored-by: Patrick von Platen <[email protected]>

Author: yiyixuxu

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py

@@ -517,7 +517,8 @@ def __call__(
         >>> pipe = StableDiffusionDepth2ImgPipeline.from_pretrained(
         ...     "stabilityai/stable-diffusion-2-depth",
         ...     torch_dtype=torch.float16,
-        ... ).to("cuda")
+        ... )
+        >>> pipe.to("cuda")
 
 
         >>> url = "http://images.cocodataset.org/val2017/000000039769.jpg"