Merge branch 'save' into trt_10

pytorch · Mar 20, 2024 · 6b86917 · 6b86917
2 parents a839c5a + 4ae6ab9
commit 6b86917
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tests/py/dynamo/models/test_models_export.py b/tests/py/dynamo/models/test_models_export.py
@@ -110,11 +110,6 @@ def test_bert_base_uncased(ir):
     model = BertModel.from_pretrained("bert-base-uncased").cuda().eval()
     input = torch.randint(0, 1, (1, 14), dtype=torch.int32).to("cuda")
     input2 = torch.randint(0, 1, (1, 14), dtype=torch.int32).to("cuda")
-    # model = (
-    #     transformers_trace(model, input_names=["input_ids", "attention_mask"])
-    #     .eval()
-    #     .cuda()
-    # )
 
     compile_spec = {
         "inputs": [