j[NddlmZddlZddlmZddlZddlmZmZ dddZ dS)) annotationsN)Path)LOGGER TORCH_VERSIONFmodeltorch.nn.Moduleim torch.Tensor output_file Path | stroptimizeboolmetadata dict | Noneprefixstrreturnctjd|dtdt|}tj||d}dtj|pii}|r>tj|ddd l m }|| || n| || |S) a/Export a PyTorch model to TorchScript format. Args: model (torch.nn.Module): The PyTorch model to export (may be NMS-wrapped). im (torch.Tensor): Example input tensor for tracing. output_file (Path | str): Path to save the exported TorchScript model. optimize (bool): Whether to optimize for mobile deployment. metadata (dict | None): Optional metadata to embed in the TorchScript archive. prefix (str): Prefix for log messages. Returns: (str): Path to the exported ``.torchscript`` file.  z starting export with torch z...F)strictz config.txtz optimizing for mobile...r)optimize_for_mobile) _extra_files) rinforrtorchjittracejsondumpstorch.utils.mobile_optimizerr_save_for_lite_interpretersave) rr r rrrts extra_filesrs g/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/ultralytics/utils/export/torchscript.pytorch2torchscriptr& s* KKVKKKKKLLLk""K 5 1 1BHN!;!;r,s#"""""  33333333  !!!!!!!r'