Skip to content

Commit f1431da

Browse files
authored
[DOCS] Port extensions command (#26863)
### Details: - *item1* - *...* ### Tickets: - *ticket-id* ### Details: - *item1* - *...* ### Tickets: - *ticket-id*
1 parent 9201c57 commit f1431da

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/articles_en/assets/snippets/ov_extensions.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ def conversion(node):
4747
#! [py_frontend_extension_aten_hardtanh]
4848
import torch
4949
from openvino.frontend import ConversionExtension, NodeContext
50-
from openvino.tools.mo import convert_model
50+
from openvino import convert_model
5151

5252

5353
class HardTanh(torch.nn.Module):
@@ -69,5 +69,5 @@ def convert_hardtanh(node: NodeContext):
6969

7070
model = HardTanh(min_val=0.1, max_val=2.0)
7171
hardtanh_ext = ConversionExtension("aten::hardtanh", convert_hardtanh)
72-
ov_model = convert_model(input_model=model, extensions=[hardtanh_ext])
72+
ov_model = convert_model(input_model=model, extension=[hardtanh_ext])
7373
#! [py_frontend_extension_aten_hardtanh]

0 commit comments

Comments
 (0)