diff --git a/examples/recipes/xnnpack_optimization/models.py b/examples/recipes/xnnpack_optimization/models.py index 42e8f24e21d..ceb6f0b796e 100644 --- a/examples/recipes/xnnpack_optimization/models.py +++ b/examples/recipes/xnnpack_optimization/models.py @@ -24,4 +24,5 @@ class OptimizationOptions(object): ), # TODO[T163161310]: takes a long time to export to exec prog and save inception_v4 quantized model "w2l": OptimizationOptions(False, True), "dl3": OptimizationOptions(True, False), + "vit": OptimizationOptions(True, False), }