(self, args)
| 32 | self.noise_augmentation_max_timestep = getattr(args, "noise_augmentation_max_timestep", 0) |
| 33 | |
| 34 | def _initialize_models(self, args): |
| 35 | self.generator = WanDiffusionWrapper(**getattr(args, "model_kwargs", {}), is_causal=True) |
| 36 | self.generator.model.requires_grad_(True) |
| 37 | |
| 38 | self.text_encoder = WanTextEncoder() |
| 39 | self.text_encoder.requires_grad_(False) |
| 40 | |
| 41 | self.vae = WanVAEWrapper() |
| 42 | self.vae.requires_grad_(False) |
| 43 | |
| 44 | def generator_loss( |
| 45 | self, |
nothing calls this directly
no test coverage detected