MCPcopy Index your code
hub / github.com/VisionXLab/OF-Diff / make_zero_conv

Method make_zero_conv

cldm/cldm.py:290–291  ·  view source on GitHub ↗
(self, channels)

Source from the content-addressed store, hash-verified

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)

Callers 1

__init__Method · 0.95

Calls 3

zero_moduleFunction · 0.90
conv_ndFunction · 0.90

Tested by

no test coverage detected