MCPcopy Create free account
hub / github.com/MotrixLab/ViMoGen / custom_forward

Method custom_forward

models/transformer/wan/modules/tm2m_model.py:673–674  ·  view source on GitHub ↗
(*inputs, **kwargs)

Source from the content-addressed store, hash-verified

671 """
672 def create_custom_forward(module):
673 def custom_forward(*inputs, **kwargs):
674 return module(*inputs, **kwargs)
675 return custom_forward
676
677 if attend_to_text_mask is None:

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected