Skip to content

Commit 7a633be

Browse files
Update onnxruntime/test/python/transformers/test_phi_vision.py
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
1 parent 1cec7b6 commit 7a633be

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

onnxruntime/test/python/transformers/test_phi_vision.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -277,7 +277,13 @@ def test_phi_vision_attention_with_mask(self):
277277
original_model = onnx.load(os.path.join(os.path.dirname(__file__), "export.onnx"))
278278
options = FusionOptions("clip")
279279
optimized_model = optimize_model(
280-
original_model, model_type="clip", num_heads=2, hidden_size=20, optimization_options=options, opt_level=0, use_gpu=True
280+
original_model,
281+
model_type="clip",
282+
num_heads=2,
283+
hidden_size=20,
284+
optimization_options=options,
285+
opt_level=0,
286+
use_gpu=True,
281287
)
282288
self.verify_fusion(optimized_model, "phi-4-v-instruct-vision-attention.onnx")
283289

0 commit comments

Comments
 (0)