fix test_stable_diffusion_diffedit

PaddlePaddle · Oct 17, 2024 · ef0c2e9 · ef0c2e9
1 parent 2cd8763
commit ef0c2e9
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/ppdiffusers/examples/dreambooth/README_sd3.md b/ppdiffusers/examples/dreambooth/README_sd3.md
@@ -131,7 +131,7 @@ import paddle
 pipe = StableDiffusion3Pipeline.from_pretrained(
     "stabilityai/stable-diffusion-3-medium-diffusers", paddle_dtype=paddle.float16
 )
-pipeline.load_lora_weights('your-lora-checkpoint')
+pipe.load_lora_weights('your-lora-checkpoint')
 
 image = pipe("A picture of a sks dog in a bucket", num_inference_steps=25).images[0]
 image.save("sks_dog_dreambooth_lora.png")

diff --git a/ppdiffusers/examples/kandinsky2_2/text_to_image/README.md b/ppdiffusers/examples/kandinsky2_2/text_to_image/README.md
@@ -124,7 +124,7 @@ prior_components = {"prior_" + k: v for k,v in pipe_prior.components.items()}
 pipe = KandinskyV22CombinedPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", **prior_components)
 
 prompt='A robot pokemon, 4k photo'
-images = pipe(prompt=prompt, negative_prompt=negative_prompt).images
+images = pipe(prompt=prompt).images
 images[0]
 ```
 

diff --git a/ppdiffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion_diffedit.py b/ppdiffusers/tests/pipelines/stable_diffusion_2/test_stable_diffusion_diffedit.py
@@ -264,7 +264,7 @@ def setUpClass(cls):
     def test_stable_diffusion_diffedit_full(self):
         generator = paddle.Generator().manual_seed(seed=0)
         pipe = StableDiffusionDiffEditPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-2", safety_checker=None, paddle_dtype="float16"
+            "stabilityai/stable-diffusion-2", safety_checker=None, paddle_dtype=paddle.float16
         )
         pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
         pipe.inverse_scheduler = DDIMInverseScheduler.from_config(pipe.scheduler.config)
@@ -299,7 +299,7 @@ def test_stable_diffusion_diffedit_full(self):
     def test_stable_diffusion_diffedit_dpm(self):
         generator = paddle.Generator().manual_seed(seed=0)
         pipe = StableDiffusionDiffEditPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-2", safety_checker=None, torch_dtype="float16"
+            "stabilityai/stable-diffusion-2", safety_checker=None, paddle_dtype=paddle.float16
         )
         pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
         pipe.inverse_scheduler = DPMSolverMultistepInverseScheduler.from_config(pipe.scheduler.config)