Spaces:
Running
Running
udpate optimum openvino export
Browse files
export.py
CHANGED
|
@@ -17,8 +17,6 @@ from optimum.exporters.onnx import validate_model_outputs
|
|
| 17 |
from optimum.exporters.tasks import TasksManager
|
| 18 |
from transformers import AutoConfig, AutoTokenizer, is_torch_available
|
| 19 |
from optimum.intel.openvino import (
|
| 20 |
-
_HEAD_TO_AUTOMODELS,
|
| 21 |
-
_TASK_ALIASES,
|
| 22 |
OVModelForAudioClassification,
|
| 23 |
OVModelForCausalLM,
|
| 24 |
OVModelForFeatureExtraction,
|
|
@@ -30,7 +28,8 @@ from optimum.intel.openvino import (
|
|
| 30 |
OVModelForTokenClassification,
|
| 31 |
OVStableDiffusionPipeline,
|
| 32 |
)
|
| 33 |
-
|
|
|
|
| 34 |
|
| 35 |
SPACES_URL = "https://huggingface.co/spaces/echarlaix/openvino-export"
|
| 36 |
|
|
|
|
| 17 |
from optimum.exporters.tasks import TasksManager
|
| 18 |
from transformers import AutoConfig, AutoTokenizer, is_torch_available
|
| 19 |
from optimum.intel.openvino import (
|
|
|
|
|
|
|
| 20 |
OVModelForAudioClassification,
|
| 21 |
OVModelForCausalLM,
|
| 22 |
OVModelForFeatureExtraction,
|
|
|
|
| 28 |
OVModelForTokenClassification,
|
| 29 |
OVStableDiffusionPipeline,
|
| 30 |
)
|
| 31 |
+
from optimum.intel.utils.constant import _TASK_ALIASES
|
| 32 |
+
from optimum.intel.openvino.utils import _HEAD_TO_AUTOMODELS
|
| 33 |
|
| 34 |
SPACES_URL = "https://huggingface.co/spaces/echarlaix/openvino-export"
|
| 35 |
|