SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix test_modeling_mpt typo in model id (#25606)

Browse Source 
Fix model id in get_large_model_config on file test_modeling_mpt
This commit is contained in:
Francisco Kurucz
2023-08-21 06:11:21 -03:00
committed by GitHub
parent f09db47a71
commit 2f8acfea1c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/mpt/test_modeling_mpt.py
View File

@@ -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)