"...zh_cn/git@developer.sourcefind.cn:OpenDAS/opencompass.git" did not exist on "f256abffd38e848211c208ac706a2e35b1ac2e94"
Unverified Commit 2f8acfea authored by Francisco Kurucz's avatar Francisco Kurucz Committed by GitHub
Browse files

Fix test_modeling_mpt typo in model id (#25606)

Fix model id in get_large_model_config on file test_modeling_mpt
parent f09db47a
...@@ -95,7 +95,7 @@ class MptModelTester: ...@@ -95,7 +95,7 @@ class MptModelTester:
self.pad_token_id = vocab_size - 1 self.pad_token_id = vocab_size - 1
def get_large_model_config(self): 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): def prepare_config_and_inputs(self, gradient_checkpointing=False):
input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size) input_ids = ids_tensor([self.batch_size, self.seq_length], self.vocab_size)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment