Update sample.py

Former-commit-id: 6e09fc266a626f465109abc199d9fb452657929e
2024-03-12 14:10:10 +08:00
parent 61306c7510
commit f8ca07a1d8
1 changed files with 1 additions and 1 deletions
@@ -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: