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

Function freqs_pixel

k_diffusion/models/axial_rope.py:71–75  ·  view source on GitHub ↗
(max_freq=10.0)

Source from the content-addressed store, hash-verified

69
70
71def freqs_pixel(max_freq=10.0):
72 def init(shape):
73 freqs = torch.linspace(1.0, max_freq / 2, shape[-1]) * math.pi
74 return freqs.log().expand(shape)
75 return init
76
77
78def freqs_pixel_log(max_freq=10.0):

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected