(self, block_scale=None, block_size=None, min_snr_value=None, fixed_frames=0, **kwargs)
| 64 | |
| 65 | class VideoDiffusionLoss(StandardDiffusionLoss): |
| 66 | def __init__(self, block_scale=None, block_size=None, min_snr_value=None, fixed_frames=0, **kwargs): |
| 67 | self.fixed_frames = fixed_frames |
| 68 | self.block_scale = block_scale |
| 69 | self.block_size = block_size |
| 70 | self.min_snr_value = min_snr_value |
| 71 | super().__init__(**kwargs) |
| 72 | |
| 73 | def __call__(self, network, denoiser, conditioner, input, batch): |
| 74 | cond = conditioner(batch) |