echarlaix HF staff commited on
Commit
1f08499
·
1 Parent(s): d45d715

Update space for latest optimum release

Browse files
Files changed (2) hide show
  1. export.py +4 -0
  2. requirements.txt +2 -2
export.py CHANGED
@@ -52,6 +52,10 @@ def convert_openvino(model_id: str, task: str, folder: str) -> List:
52
  task = _TASK_ALIASES.get(task, task)
53
  if task not in _HEAD_TO_AUTOMODELS:
54
  raise ValueError(f"The task '{task}' is not supported, only {_HEAD_TO_AUTOMODELS.keys()} tasks are supported")
 
 
 
 
55
  auto_model_class = eval(_HEAD_TO_AUTOMODELS[task])
56
  ov_model = auto_model_class.from_pretrained(model_id, export=True)
57
  ov_model.save_pretrained(folder)
 
52
  task = _TASK_ALIASES.get(task, task)
53
  if task not in _HEAD_TO_AUTOMODELS:
54
  raise ValueError(f"The task '{task}' is not supported, only {_HEAD_TO_AUTOMODELS.keys()} tasks are supported")
55
+
56
+ if task == "text2text-generation":
57
+ raise ValueError("Export of Seq2Seq models is currently disabled.")
58
+
59
  auto_model_class = eval(_HEAD_TO_AUTOMODELS[task])
60
  ov_model = auto_model_class.from_pretrained(model_id, export=True)
61
  ov_model.save_pretrained(folder)
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  transformers
2
  diffusers
3
- optimum==1.7.3
4
- git+https://github.com/huggingface/optimum-intel.git#egg=optimum-intel[openvino]
 
1
  transformers
2
  diffusers
3
+ optimum==1.8.1
4
+ optimum-intel[openvino]==1.8.0