fbcode/aitemplate/AITemplate/examples/03_bert

Reviewed By: hl475

Differential Revision: D80779735

Author: CodemodService CodemodConfigDisabledTests

examples/03_bert/test_correctness.py

@@ -32,27 +32,6 @@ class BertBaseUncasedTest(unittest.TestCase):
     def setUpClass(cls) -> None:
         torch.manual_seed(0)
 
-    def test_bert_base_uncased(self):
-        model_path = "bert-base-uncased"
-        if ManifoldClient is not None:
-            model_path = "/tmp/aitemplate_bert/bert-base-uncased"
-            os.makedirs(model_path, exist_ok=True)
-            with ManifoldClient.get_client(bucket="glow_test_data") as client:
-                await_sync(
-                    client.getRecursive(
-                        manifold_path="tree/aitemplate/bert/bert-base-uncased",
-                        local_path=model_path,
-                    )
-                )
-        run_model(
-            prompt="The quick brown fox jumps over the lazy dog.",
-            activation="fast_gelu",
-            graph_mode=True,
-            use_fp16_acc=True,
-            verify=True,
-            model_path=model_path,
-        )
-
 
 if __name__ == "__main__":
     unittest.main()