From ac5350f778619c7c59f87438c1cec595c5047f09 Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 7 Mar 2024 00:07:14 -0500 Subject: [PATCH] Another fix --- torchbenchmark/models/BERT_pytorch/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchbenchmark/models/BERT_pytorch/__init__.py b/torchbenchmark/models/BERT_pytorch/__init__.py index f041ea54df..298015c577 100644 --- a/torchbenchmark/models/BERT_pytorch/__init__.py +++ b/torchbenchmark/models/BERT_pytorch/__init__.py @@ -208,7 +208,7 @@ def get_module(self): return self.model.model, self.example_inputs def set_module(self, new_model): - self.model.bert = new_model + self.model.model = new_model def eval(self) -> typing.Tuple[torch.Tensor]: model = self.model