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

Function apply_rotary_emb_

k_diffusion/models/modules.py:244–245  ·  view source on GitHub ↗
(x, theta)

Source from the content-addressed store, hash-verified

242
243
244def apply_rotary_emb_(x, theta):
245 return ApplyRotaryEmbeddingInplace.apply(x, theta, False)
246
247
248class AxialRoPE(nn.Module):

Callers 4

forwardMethod · 0.90
forwardMethod · 0.85
forwardMethod · 0.85
forwardMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected