| 65 | |
| 66 | |
| 67 | class DiffusionUncond(nn.Module): |
| 68 | def __init__(self, global_args): |
| 69 | super().__init__() |
| 70 | |
| 71 | self.diffusion = DiffusionAttnUnet1D(global_args, n_attn_layers=4) |
| 72 | self.diffusion_ema = deepcopy(self.diffusion) |
| 73 | self.rng = torch.quasirandom.SobolEngine(1, scramble=True) |
| 74 | |
| 75 | |
| 76 | def download(model_name): |
no outgoing calls
no test coverage detected
searching dependent graphs…