Skip to content

GitLab

"...transforms/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "edde825516e1ee087794cabadda90fcefd5fe393"
Commit ffe7b93b authored by anton-l's avatar anton-l
Browse files

DDIM resolution->image_size

parent f82ebb9a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/pipelines/ddim/pipeline_ddim.py
View file @ ffe7b93b
...@@ -36,7 +36,7 @@ class DDIMPipeline(DiffusionPipeline): ...@@ -36,7 +36,7 @@ class DDIMPipeline(DiffusionPipeline):
# Sample gaussian noise to begin loop # Sample gaussian noise to begin loop
image = torch.randn( image = torch.randn(
(batch_size, self.unet.in_channels, self.unet.resolution, self.unet.resolution), (batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size),
generator=generator, generator=generator,
) )
image = image.to(torch_device) image = image.to(torch_device)
......
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