From 438d06c95acdee6c94f6ba144263f763c2d338a0 Mon Sep 17 00:00:00 2001 From: Francisco Kurucz Date: Tue, 6 Aug 2024 07:48:32 -0300 Subject: [PATCH] Fix get large model config for Switch Transformer encoder only tester (#32438) --- .../switch_transformers/test_modeling_switch_transformers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/switch_transformers/test_modeling_switch_transformers.py b/tests/models/switch_transformers/test_modeling_switch_transformers.py index 13241151a8..13215b2826 100644 --- a/tests/models/switch_transformers/test_modeling_switch_transformers.py +++ b/tests/models/switch_transformers/test_modeling_switch_transformers.py @@ -770,7 +770,7 @@ class SwitchTransformersEncoderOnlyModelTester: self.is_training = is_training 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): input_ids = ids_tensor([self.batch_size, self.encoder_seq_length], self.vocab_size)