"tests/pipelines/vscode:/vscode.git/clone" did not exist on "34c90dbb31f4956e72086ac43ca7d8154c2aadae"
Unverified Commit 1c96f82e authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Update one_step_unet.py

Fix dummy community pipeline
parent ce144d6d
......@@ -12,7 +12,7 @@ class UnetSchedulerOneForwardPipeline(DiffusionPipeline):
def __call__(self):
image = torch.randn(
(1, self.unet.in_channels, self.unet.sample_size, self.unet.sample_size),
(1, self.unet.config.in_channels, self.unet.config.sample_size, self.unet.config.sample_size),
)
timestep = 1
......
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