MCPcopy Create free account
hub / github.com/VisionXLab/OF-Diff / count_flops

Method count_flops

ldm/modules/diffusionmodules/openaimodel.py:410–411  ·  view source on GitHub ↗
(model, _x, y)

Source from the content-addressed store, hash-verified

408
409 @staticmethod
410 def count_flops(model, _x, y):
411 return count_flops_attn(model, _x, y)
412
413
414class UNetModel(nn.Module):

Callers

nothing calls this directly

Calls 1

count_flops_attnFunction · 0.85

Tested by

no test coverage detected