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

Method count_flops

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

Source from the content-addressed store, hash-verified

376
377 @staticmethod
378 def count_flops(model, _x, y):
379 return count_flops_attn(model, _x, y)
380
381
382class QKVAttention(nn.Module):

Callers

nothing calls this directly

Calls 1

count_flops_attnFunction · 0.85

Tested by

no test coverage detected