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

Fix get large model config for Switch Transformer encoder only tester (#32438)

Browse Source
This commit is contained in:
Francisco Kurucz
2024-08-06 07:48:32 -03:00
committed by GitHub
parent fb66ef8147
commit 438d06c95a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/switch_transformers/test_modeling_switch_transformers.py
View File

@@ -770,7 +770,7 @@ class SwitchTransformersEncoderOnlyModelTester:
self.is_training = is_training self.is_training = is_training
def get_large_model_config(self): def get_large_model_config(self):
return SwitchTransformersConfig.from_pretrained("switch_base_8") return SwitchTransformersConfig.from_pretrained("google/switch-base-8")
def prepare_config_and_inputs(self): def prepare_config_and_inputs(self):
input_ids = ids_tensor([self.batch_size, self.encoder_seq_length], self.vocab_size) input_ids = ids_tensor([self.batch_size, self.encoder_seq_length], self.vocab_size)