Function half

wan/modules/attention.py:68–69 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

66	b, lq, lk, out_dtype = q.size(0), q.size(1), k.size(1), q.dtype
67
68	def half(x):
69	return x if x.dtype in half_dtypes else x.to(dtype)
70
71	# preprocess query
72	if q_lens is None:

flash_attentionFunction · 0.70

no outgoing calls

no test coverage detected