Skip to content

Commit 4f36b6f

Browse files
committed
code style
1 parent 0b9b253 commit 4f36b6f

File tree

2 files changed

+9
-3
lines changed

2 files changed

+9
-3
lines changed

optimum/exporters/openvino/model_configs.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,13 @@
1919
from transformers.utils import is_tf_available
2020

2121
from optimum.exporters.onnx.config import TextDecoderOnnxConfig, TextDecoderWithPositionIdsOnnxConfig
22-
from optimum.exporters.onnx.model_configs import FalconOnnxConfig, GemmaOnnxConfig, LlamaOnnxConfig, MPTOnnxConfig, PhiOnnxConfig
22+
from optimum.exporters.onnx.model_configs import (
23+
FalconOnnxConfig,
24+
GemmaOnnxConfig,
25+
LlamaOnnxConfig,
26+
MPTOnnxConfig,
27+
PhiOnnxConfig,
28+
)
2329
from optimum.exporters.tasks import TasksManager
2430
from optimum.utils import DEFAULT_DUMMY_SHAPES
2531
from optimum.utils.input_generators import (
@@ -458,7 +464,7 @@ def patch_model_for_export(
458464
return MPTModelPatcher(self, model, model_kwargs=model_kwargs)
459465

460466

461-
@register_in_tasks_manager(
467+
@register_in_tasks_manager(
462468
"phi3",
463469
*[
464470
"feature-extraction",

tests/openvino/test_modeling.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -553,7 +553,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
553553
"internlm2",
554554
"olmo",
555555
"orion",
556-
"phi3"
556+
"phi3",
557557
)
558558

559559
@parameterized.expand(SUPPORTED_ARCHITECTURES)

0 commit comments

Comments
 (0)