MCPcopy Create free account
hub / github.com/CompVis/diff2flow / count_flops

Method count_flops

diff2flow/models/unet/openaimodel.py:412–413  ·  view source on GitHub ↗
(model, _x, y)

Source from the content-addressed store, hash-verified

410
411 @staticmethod
412 def count_flops(model, _x, y):
413 return count_flops_attn(model, _x, y)
414
415
416class Timestep(nn.Module):

Callers

nothing calls this directly

Calls 1

count_flops_attnFunction · 0.85

Tested by

no test coverage detected