| 44 | |
| 45 | |
| 46 | class CausalResnetBlock1D(ResnetBlock1D): |
| 47 | def __init__(self, dim: int, dim_out: int, time_emb_dim: int, groups: int = 8): |
| 48 | super(CausalResnetBlock1D, self).__init__(dim, dim_out, time_emb_dim, groups) |
| 49 | self.block1 = CausalBlock1D(dim, dim_out) |
| 50 | self.block2 = CausalBlock1D(dim_out, dim_out) |
| 51 | |
| 52 | class CausalConv1d(torch.nn.Conv1d): |
| 53 | def __init__( |
nothing calls this directly
no outgoing calls
no test coverage detected