Skip to content

Commit

Permalink
[DOCS] Port extensions command (#26863)
Browse files Browse the repository at this point in the history
### Details:
 - *item1*
 - *...*

### Tickets:
 - *ticket-id*

### Details:
 - *item1*
 - *...*

### Tickets:
 - *ticket-id*
  • Loading branch information
tsavina authored Sep 30, 2024
1 parent 9201c57 commit f1431da
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/articles_en/assets/snippets/ov_extensions.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,7 @@ def conversion(node):
#! [py_frontend_extension_aten_hardtanh]
import torch
from openvino.frontend import ConversionExtension, NodeContext
from openvino.tools.mo import convert_model
from openvino import convert_model


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

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

0 comments on commit f1431da

Please sign in to comment.