Skip to content

GitLab

Commit a3784522 authored by patil-suraj's avatar patil-suraj
Browse files

fix initial image in ddim

parent d4c2bcf8
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/pipeline_latent_diffusion.py
View file @ a3784522
...@@ -900,7 +900,7 @@ class LatentDiffusion(DiffusionPipeline): ...@@ -900,7 +900,7 @@ class LatentDiffusion(DiffusionPipeline):
num_trained_timesteps = self.noise_scheduler.timesteps num_trained_timesteps = self.noise_scheduler.timesteps
inference_step_times = range(0, num_trained_timesteps, num_trained_timesteps // num_inference_steps) inference_step_times = range(0, num_trained_timesteps, num_trained_timesteps // num_inference_steps)
image = self.noise_scheduler.sample_noise( image = torch.randn(
(batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size), (batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size),
device=torch_device, device=torch_device,
generator=generator, generator=generator,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment