a dk @s<dZddlmZddlmZddlmZmZm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZddlmZmZddl m!Z!dd l m"Z"dd lm#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+gd Z,d e_-d e!_-d Z.ej/Z0e j1ddddddZ2ej3Z4ddddZ5ddddZ6ej7Z8ej9Z:dS)zONNX exporter.)_C)_onnx)_CAFFE2_ATEN_FALLBACKOperatorExportTypesTensorProtoDataType TrainingMode) _deprecationerrorssymbolic_caffe2symbolic_helpersymbolic_opset7symbolic_opset8symbolic_opset9symbolic_opset10symbolic_opset11symbolic_opset12symbolic_opset13symbolic_opset14symbolic_opset15symbolic_opset16symbolic_opset17symbolic_opset18utils) ExportTypesSymbolicContext) JitScalarType) CheckerError) _optimize_graph_run_symbolic_function_run_symbolic_methodexportexport_to_pretty_stringis_in_onnx_exportregister_custom_op_symbolicselect_model_mode_for_exportunregister_custom_op_symbolic)r rr r r rrrrrrrrrrrrrrrrr!r"r#r%r$r& disable_log enable_logrz torch.onnxZpytorchz1.12.0z2.0zuse `torch.onnx.export` instead)ZsinceZ removed_inZ instructionscOstj|i|S)N)r_export)argskwargsr,\/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/onnx/__init__.pyr)^sr)N)returncCstddS)zEnables ONNX logging.TNrZ_jit_set_onnx_log_enabledr,r,r,r-r(ksr(cCstddS)zDisables ONNX logging.FNr/r,r,r,r-r'psr');__doc__ZtorchrZtorch._CrZ_C_onnxZtorch._C._onnxrrrrr r r r r rrrrrrrrrrrrZ_exporter_statesrrZ _type_utilsrrrrr r!r"r#r$r%r&__all__ __module__Z producer_nameZPRODUCER_VERSIONZproducer_version deprecatedr)Z_jit_is_onnx_log_enabledZis_onnx_log_enabledr(r'Z_jit_set_onnx_log_output_streamZset_log_streamZ _jit_onnx_loglogr,r,r,r-s,  L  , '