neural-os

Runtime error

yuntian-deng commited on Oct 29, 2024

Commit

bc03c32

1 Parent(s): f0231ae

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -39,7 +39,7 @@ def sample_frame(model: LatentDiffusion, prompt: str, image_sequence: torch.Tens
         c = model.get_learned_conditioning(c_dict)
         c = model.enc_concat_seq(c, c_dict, 'c_concat')
-        samples_ddim, _ = sampler.sample(S=200,
                                          conditioning=c,
                                          batch_size=1,
                                          shape=[3, 64, 64],

         c = model.get_learned_conditioning(c_dict)
         c = model.enc_concat_seq(c, c_dict, 'c_concat')
+        samples_ddim, _ = sampler.sample(S=4,
                                          conditioning=c,
                                          batch_size=1,
                                          shape=[3, 64, 64],