Method sample

show-o2/models/wan21_vae.py:654–661

(self, videos, deterministic=False, return_features=False)

            return self.model.decode(zs, self.scale).float().clamp_(-1, 1)

    def sample(self, videos, deterministic=False, return_features=False):
        with amp.autocast(dtype=self.dtype):
            if return_features:
                out, feats = self.model.sample(videos, self.scale, deterministic=deterministic,
                                               return_features=return_features)
                return out.float(), feats.float()
            else:
                return self.model.sample(videos, self.scale, deterministic=deterministic).float()

    def per_decode(self, zs):
        outputs = []