MCPcopy Create free account
hub / github.com/Meshcapade/difflocks / forward

Method forward

k_diffusion/models/modules.py:550–552  ·  view source on GitHub ↗
(self, x, pos, cond)

Source from the content-addressed store, hash-verified

548 self.ff = FeedForwardBlock(d_model, d_ff, cond_features, dropout=dropout)
549
550 def forward(self, x, pos, cond):
551 x = checkpoint(self.ff, x, cond)
552 return x
553
554
555class Level(nn.ModuleList):

Callers

nothing calls this directly

Calls 1

checkpointFunction · 0.85

Tested by

no test coverage detected