update phi default onnx opset

echarlaix · echarlaix · commit a809853d4042 · 2024-03-21T16:42:46.000+01:00
diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py
@@ -252,6 +252,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig):
 
 
 class PhiOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):
+    DEFAULT_ONNX_OPSET = 14  # Phi now uses F.scaled_dot_product_attention by default for torch>=2.1.1.
     NORMALIZED_CONFIG_CLASS = NormalizedTextConfig
 
 

Original file line number	Diff line number	Diff line change
`@@ -252,6 +252,7 @@ class GemmaOnnxConfig(LlamaOnnxConfig):`
`252`	`252`
`253`	`253`
`254`	`254`	`class PhiOnnxConfig(TextDecoderWithPositionIdsOnnxConfig):`
	`255`	`+ DEFAULT_ONNX_OPSET = 14 # Phi now uses F.scaled_dot_product_attention by default for torch>=2.1.1.`
`255`	`256`	`NORMALIZED_CONFIG_CLASS = NormalizedTextConfig`
`256`	`257`
`257`	`258`