comment
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -940,7 +940,7 @@ class StableDiffusionControlNetImg2ImgPipeline(DiffusionPipeline, TextualInversi
|
|
940 |
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
|
941 |
latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)
|
942 |
|
943 |
-
6. Prepare latent variables
|
944 |
latents = self.prepare_latents(
|
945 |
image,
|
946 |
latent_timestep,
|
|
|
940 |
timesteps, num_inference_steps = self.get_timesteps(num_inference_steps, strength, device)
|
941 |
latent_timestep = timesteps[:1].repeat(batch_size * num_images_per_prompt)
|
942 |
|
943 |
+
# 6. Prepare latent variables
|
944 |
latents = self.prepare_latents(
|
945 |
image,
|
946 |
latent_timestep,
|