MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / QKVAttentionLegacy

Class QKVAttentionLegacy

guided_diffusion/unet.py:393–423  ·  view source on GitHub ↗

A module which performs QKV attention. Matches legacy QKVAttention + input/ouput heads shaping

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

__init__Method · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected