in optimum/exporters/onnx/__main__.py [0:0]
def main_export(
model_name_or_path: str,
output: Union[str, Path],
task: str = "auto",
opset: Optional[int] = None,
device: str = "cpu",
dtype: Optional[str] = None,
fp16: Optional[bool] = False,
optimize: Optional[str] = None,
monolith: bool = False,
no_post_process: bool = False,
framework: Optional[str] = None,
atol: Optional[float] = None,
pad_token_id: Optional[int] = None,
# hub options
subfolder: str = "",
revision: str = "main",
force_download: bool = False,
local_files_only: bool = False,
trust_remote_code: bool = False,
cache_dir: str = HUGGINGFACE_HUB_CACHE,
token: Optional[Union[bool, str]] = None,
########################################
for_ort: bool = False,
do_validation: bool = True,
model_kwargs: Optional[Dict[str, Any]] = None,
custom_onnx_configs: Optional[Dict[str, "OnnxConfig"]] = None,
fn_get_submodels: Optional[Callable] = None,
use_subprocess: bool = False,
_variant: str = "default",
library_name: Optional[str] = None,
legacy: bool = False,
no_dynamic_axes: bool = False,
do_constant_folding: bool = True,
slim: bool = False,
**kwargs_shapes,