def create_custom_forward(module):

    def custom_forward(*inputs):
        return module(*inputs, use_cache, output_attentions)
    return custom_forward
def create_custom_forward(module):

    def custom_forward(*inputs):
        return module(*inputs, use_cache, output_attentions)
    return custom_forward