(self, inputs)
| 131 | |
| 132 | |
| 133 | def noise_initializer(self, inputs): |
| 134 | length = (inputs["num_frames"] - 1) // 4 + 1 + 1 |
| 135 | shape = (1, self.pipe.vae.model.z_dim, length, inputs["height"] // self.pipe.vae.upsampling_factor, inputs["width"] // self.pipe.vae.upsampling_factor) |
| 136 | noise = self.pipe.generate_noise(shape, seed=None, rand_device=inputs["rand_device"]) |
| 137 | noise = torch.concat((noise[:, :, -1:], noise[:, :, :-1]), dim=2) |
| 138 | inputs["noise"] = noise |
| 139 | return inputs |
| 140 | |
| 141 | |
| 142 | def prompt_embedder(self, inputs): |
no test coverage detected