"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "a413c725d40027134f28f92974ad7d61751f5640"
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
......@@ -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)
......
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