From f8ca07a1d81fb5d372e8f74b882d3eff06174689 Mon Sep 17 00:00:00 2001 From: lb203 <62638829+LinB203@users.noreply.github.com> Date: Tue, 12 Mar 2024 14:10:10 +0800 Subject: [PATCH] Update sample.py Former-commit-id: 6e09fc266a626f465109abc199d9fb452657929e --- opensora/sample/sample.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/opensora/sample/sample.py b/opensora/sample/sample.py index 0bc0dc7bf..9f127b085 100644 --- a/opensora/sample/sample.py +++ b/opensora/sample/sample.py @@ -69,7 +69,7 @@ def main(args): bar = tqdm(range(args.num_sample)) for i in bar: # Create sampling noise: - z = torch.randn(1, args.num_frames // ae_stride_config[args.ae][0], model.in_channels, latent_size[0], latent_size[1], device=device) + z = torch.randn(1, args.num_frames // ae_stride_config[args.ae][0], model.module.in_channels, latent_size[0], latent_size[1], device=device) # Setup classifier-free guidance: if using_cfg: