We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5b17ca3 commit 511855aCopy full SHA for 511855a
optimum/exporters/openvino/convert.py
@@ -23,7 +23,6 @@
23
from transformers.generation import GenerationMixin
24
from transformers.utils import is_tf_available, is_torch_available
25
26
-from openvino.frontend.pytorch.ts_decoder import TorchScriptPythonDecoder
27
from openvino.runtime import Model, save_model
28
from openvino.runtime.exceptions import OVTypeError
29
from openvino.tools.ovc import convert_model
@@ -367,6 +366,7 @@ def export_pytorch(
367
366
import torch
368
from torch.utils._pytree import tree_map
369
+ from openvino.frontend.pytorch.ts_decoder import TorchScriptPythonDecoder
370
from optimum.exporters.utils import check_dummy_inputs_are_allowed
371
372
logger.info(f"Using framework PyTorch: {torch.__version__}")
0 commit comments