experimental_experiment.torch_interpreter.onnx_export_errors

experimental_experiment.torch_interpreter.onnx_export_errors.bypass_export_some_errors()

Tries to bypass some functions torch.export.export() does not support:

• torch.jit.isinstance

• torch._dynamo.mark_static_address

• Serialialization of MambaCache (in transformers)

• Serialialization of DynamicCache (in transformers)

Serialization issues happen when a module takes one input or output has a type torch.export.export() cannot serialize.