File tree 1 file changed +16
-1
lines changed
optimum/exporters/openvino
1 file changed +16
-1
lines changed Original file line number Diff line number Diff line change 19
19
from transformers .utils import is_tf_available
20
20
21
21
from optimum .exporters .onnx .config import TextDecoderOnnxConfig , TextDecoderWithPositionIdsOnnxConfig
22
- from optimum .exporters .onnx .model_configs import GemmaOnnxConfig , LlamaOnnxConfig
22
+ from optimum .exporters .onnx .model_configs import GemmaOnnxConfig , LlamaOnnxConfig , PhiOnnxConfig
23
23
from optimum .exporters .tasks import TasksManager
24
24
from optimum .utils import DEFAULT_DUMMY_SHAPES
25
25
from optimum .utils .input_generators import (
@@ -437,3 +437,18 @@ class OrionOpenVINOConfig(TextDecoderWithPositionIdsOnnxConfig):
437
437
DUMMY_INPUT_GENERATOR_CLASSES = (DummyTextInputGenerator , MistralDummyPastKeyValuesGenerator )
438
438
DUMMY_PKV_GENERATOR_CLASS = MistralDummyPastKeyValuesGenerator
439
439
NORMALIZED_CONFIG_CLASS = NormalizedTextConfig
440
+
441
+
442
+ @register_in_tasks_manager (
443
+ "phi3" ,
444
+ * [
445
+ "feature-extraction" ,
446
+ "feature-extraction-with-past" ,
447
+ "text-generation" ,
448
+ "text-generation-with-past" ,
449
+ "text-classification" ,
450
+ ],
451
+ library_name = "transformers" ,
452
+ )
453
+ class Phi3OpenVINOConfig (PhiOnnxConfig ):
454
+ pass
You can’t perform that action at this time.
0 commit comments