Skip to content

Commit 1c96f82

Browse files
Update one_step_unet.py
Fix dummy community pipeline
1 parent ce144d6 commit 1c96f82

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/community/one_step_unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def __init__(self, unet, scheduler):
1212

1313
def __call__(self):
1414
image = torch.randn(
15-
(1, self.unet.in_channels, self.unet.sample_size, self.unet.sample_size),
15+
(1, self.unet.config.in_channels, self.unet.config.sample_size, self.unet.config.sample_size),
1616
)
1717
timestep = 1
1818

0 commit comments

Comments
 (0)