"vscode:/vscode.git/clone" did not exist on "9249be28ccd7359a2c9099a37874fc7660ddfd79"
Commit a3784522 authored by patil-suraj's avatar patil-suraj
Browse files

fix initial image in ddim

parent d4c2bcf8
......@@ -900,7 +900,7 @@ class LatentDiffusion(DiffusionPipeline):
num_trained_timesteps = self.noise_scheduler.timesteps
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),
device=torch_device,
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