flow_matching.utils.model_wrapper

ModelWrapper

ModelWrapper

This class is used to wrap around another model, adding custom forward pass logic.