Skip to content

Commit 6430003

Browse files
committed
fix format
1 parent e69eac0 commit 6430003

File tree

2 files changed

+0
-2
lines changed

2 files changed

+0
-2
lines changed

optimum/exporters/openvino/__main__.py

-1
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,6 @@ def main_export(
136136
>>> main_export("gpt2", output="gpt2_onnx/")
137137
```
138138
"""
139-
140139
original_task = task
141140
task = TasksManager.map_from_synonym(task)
142141
framework = TasksManager.determine_framework(model_name_or_path, subfolder=subfolder, framework=framework)

tests/openvino/test_export.py

-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919
from typing import Optional
2020

2121
from parameterized import parameterized
22-
from transformers import AutoConfig
2322
from utils_tests import MODEL_NAMES
2423

2524
from optimum.exporters.onnx.constants import SDPA_ARCHS_ONNX_EXPORT_NOT_SUPPORTED

0 commit comments

Comments
 (0)