Skip to content

GitLab

Unverified Commit 438d06c9 authored by Francisco Kurucz's avatar Francisco Kurucz Committed by GitHub
Browse files

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

parent fb66ef81
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/switch_transformers/test_modeling_switch_transformers.py
View file @ 438d06c9
...@@ -770,7 +770,7 @@ class SwitchTransformersEncoderOnlyModelTester: ...@@ -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)
......
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