Skip to content

Commit 511855a

Browse files
committedJan 13, 2025
Move decoder input closer to the usage. For cases when torch is not available in openvino
1 parent 5b17ca3 commit 511855a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed
 

‎optimum/exporters/openvino/convert.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
from transformers.generation import GenerationMixin
2424
from transformers.utils import is_tf_available, is_torch_available
2525

26-
from openvino.frontend.pytorch.ts_decoder import TorchScriptPythonDecoder
2726
from openvino.runtime import Model, save_model
2827
from openvino.runtime.exceptions import OVTypeError
2928
from openvino.tools.ovc import convert_model
@@ -367,6 +366,7 @@ def export_pytorch(
367366
import torch
368367
from torch.utils._pytree import tree_map
369368

369+
from openvino.frontend.pytorch.ts_decoder import TorchScriptPythonDecoder
370370
from optimum.exporters.utils import check_dummy_inputs_are_allowed
371371

372372
logger.info(f"Using framework PyTorch: {torch.__version__}")

0 commit comments

Comments
 (0)