diff --git a/examples/xnnpack/__init__.py b/examples/xnnpack/__init__.py index 81404dcf6b7..d8de9f6a36e 100644 --- a/examples/xnnpack/__init__.py +++ b/examples/xnnpack/__init__.py @@ -4,6 +4,8 @@ # This source code is licensed under the BSD-style license found in the # LICENSE file in the root directory of this source tree. +# pyre-unsafe + from dataclasses import dataclass @@ -24,14 +26,14 @@ class XNNPACKOptions(object): "mv3": XNNPACKOptions(True, True), "resnet18": XNNPACKOptions(True, True), "resnet50": XNNPACKOptions(True, True), - "vit": XNNPACKOptions(False, True), # T161242362 + "vit": XNNPACKOptions(True, True), "w2l": XNNPACKOptions(True, True), "edsr": XNNPACKOptions(True, True), - "mobilebert": XNNPACKOptions(False, True), # T197452682 + "mobilebert": XNNPACKOptions(True, True), "llama2": XNNPACKOptions(False, True), "emformer_join": XNNPACKOptions(True, True), - "emformer_predict": XNNPACKOptions(False, True), # T197457838 - "emformer_transcribe": XNNPACKOptions(False, True), # T197449765 + "emformer_predict": XNNPACKOptions(True, True), + "emformer_transcribe": XNNPACKOptions(True, True), }