From 1c96f82ed99eede072606e8cc33a975bf4453e90 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Sun, 9 Apr 2023 19:22:18 +0100 Subject: [PATCH] Update one_step_unet.py Fix dummy community pipeline --- examples/community/one_step_unet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/community/one_step_unet.py b/examples/community/one_step_unet.py index f3eaf1e0eb..7d34bfd831 100755 --- a/examples/community/one_step_unet.py +++ b/examples/community/one_step_unet.py @@ -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