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

Add large model config (#9140)

Browse source
This commit is contained in:
Lysandre Debut 2020-12-15 16:03:59 -05:00 committed by GitHub
parent 2a7e8e1608
commit 0b2f46fa9e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/test_modeling_t5.py
View file

@ -610,6 +610,9 @@ class T5EncoderOnlyModelTester:
self.scope = None
self.is_training = is_training
def get_large_model_config(self):
return T5Config.from_pretrained("t5-base")
def prepare_config_and_inputs(self):
input_ids = ids_tensor([self.batch_size, self.encoder_seq_length], self.vocab_size)