(self, sample)
| 53 | self.act = nn.Mish() |
| 54 | |
| 55 | def forward(self, sample): |
| 56 | sample = self.linear(sample) |
| 57 | |
| 58 | if self.act is not None: |
| 59 | sample = self.act(sample) |
| 60 | |
| 61 | return sample |
| 62 | |
| 63 | class UNet3DConditionModel(ModelMixin, ConfigMixin): |
| 64 | _supports_gradient_checkpointing = True |
nothing calls this directly
no outgoing calls
no test coverage detected