From abd7502ccaab72d78ebf229cc670836acf55b9e4 Mon Sep 17 00:00:00 2001 From: A-Cepheus <60658915+A-Cepheus@users.noreply.github.com> Date: Thu, 4 Jan 2024 04:00:15 +0800 Subject: [PATCH] fix: typo in sa (#838) --- training/bing_bert/nvidia/modelingpreln.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/bing_bert/nvidia/modelingpreln.py b/training/bing_bert/nvidia/modelingpreln.py index a7e398e26..9856f0607 100755 --- a/training/bing_bert/nvidia/modelingpreln.py +++ b/training/bing_bert/nvidia/modelingpreln.py @@ -1041,7 +1041,7 @@ def forward(self, position_ids=None, inputs_embeds=None, pad_token_id=self.pad_token_id, - model_mbeddings=self.embeddings) + model_embeddings=self.embeddings) embedding_output = self.embeddings(input_ids, token_type_ids) encoded_layers = self.encoder(