echarlaix HF staff commited on
Commit
67f027f
1 Parent(s): d9f713b

udpate optimum openvino export

Browse files
Files changed (1) hide show
  1. export.py +2 -3
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