Skip to content

Commit 7a716f8

Browse files
authored
Add token classification for qwen2 ov export (#1299)
1 parent b7819a9 commit 7a716f8

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

optimum/exporters/openvino/model_configs.py

Lines changed: 12 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,18 @@ def patch_model_for_export(
259259
return BaichuanModelPatcher(self, model, model_kwargs=model_kwargs)
260260

261261

262-
@register_in_tasks_manager("qwen2", *["text-generation", "text-generation-with-past"], library_name="transformers")
262+
@register_in_tasks_manager(
263+
"qwen2",
264+
*[
265+
"text-generation",
266+
"text-generation-with-past",
267+
"feature-extraction",
268+
"feature-extraction-with-past",
269+
"text-classification",
270+
"token-classification",
271+
],
272+
library_name="transformers",
273+
)
263274
class Qwen2OpenVINOConfig(TextDecoderWithPositionIdsOnnxConfig):
264275
DEFAULT_ONNX_OPSET = 14
265276

0 commit comments

Comments
 (0)