(self, input_dim, out_dim, cond_dims)
| 641 | class LocalCondProj(nn.Module): |
| 642 | |
| 643 | def __init__(self, input_dim, out_dim, cond_dims): |
| 644 | super().__init__() |
| 645 | self.proj_in=(nn.Linear(input_dim, out_dim, bias=False)) |
| 646 | self.ff=FeedForwardBlock(out_dim, out_dim, cond_dims) |
| 647 | |
| 648 | |
| 649 | def forward(self, x, noise_cond): |
nothing calls this directly
no test coverage detected