saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats 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
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)