From 422fd9fe95a04e70344722a8a62fc4397ff56489 Mon Sep 17 00:00:00 2001 From: AmosLewis Date: Thu, 24 Oct 2024 13:00:41 -0700 Subject: [PATCH] Fix the coat model with onnx.if bug --- alt_e2eshark/onnx_tests/models/azure_models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/alt_e2eshark/onnx_tests/models/azure_models.py b/alt_e2eshark/onnx_tests/models/azure_models.py index c00c72cf..f14756e6 100644 --- a/alt_e2eshark/onnx_tests/models/azure_models.py +++ b/alt_e2eshark/onnx_tests/models/azure_models.py @@ -42,6 +42,8 @@ # if the model has significant shape issues, consider applying basic optimizations before import by adding to this list: basic_opt = [ + "coat_mini", + "coat_tiny", "mvitv2_base", "mvitv2_large", "mvitv2_small",