Function half

models/transformer/wan/modules/t2m_model.py:64–65 · view source on GitHub ↗

(x)

Source from the content-addressed store, hash-verified

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

flash_attentionFunction · 0.70

no outgoing calls

no test coverage detected