(self, channels)
| 288 | self._feature_size += ch |
| 289 | |
| 290 | def make_zero_conv(self, channels): |
| 291 | return TimestepEmbedSequential(zero_module(conv_nd(self.dims, channels, channels, 1, padding=0))) |
| 292 | |
| 293 | def forward(self, x, hint, timesteps, context, **kwargs): |
| 294 | t_emb = timestep_embedding(timesteps, self.model_channels, repeat_only=False) |
no test coverage detected