From 2f8acfea1ca11fe3479fb379ccbded516d0cff57 Mon Sep 17 00:00:00 2001 From: Francisco Kurucz Date: Mon, 21 Aug 2023 06:11:21 -0300 Subject: [PATCH] Fix test_modeling_mpt typo in model id (#25606) Fix model id in get_large_model_config on file test_modeling_mpt --- tests/models/mpt/test_modeling_mpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/mpt/test_modeling_mpt.py b/tests/models/mpt/test_modeling_mpt.py index 5b4af4f72..c2d3ae0d0 100644 --- a/tests/models/mpt/test_modeling_mpt.py +++ b/tests/models/mpt/test_modeling_mpt.py @@ -95,7 +95,7 @@ class MptModelTester: self.pad_token_id = vocab_size - 1 def get_large_model_config(self): - return MptConfig.from_pretrained("mosaicml/mpt-7") + return MptConfig.from_pretrained("mosaicml/mpt-7b") def prepare_config_and_inputs(self, gradient_checkpointing=False): input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size)