@@ -69,7 +69,7 @@ class FlaxOPTModelTester:
|
||||
embed_dim=16,
|
||||
word_embed_proj_dim=16,
|
||||
initializer_range=0.02,
|
||||
attn_implemetation="eager",
|
||||
attn_implementation="eager",
|
||||
):
|
||||
self.parent = parent
|
||||
self.batch_size = batch_size
|
||||
@@ -92,7 +92,7 @@ class FlaxOPTModelTester:
|
||||
self.word_embed_proj_dim = word_embed_proj_dim
|
||||
self.initializer_range = initializer_range
|
||||
self.is_encoder_decoder = False
|
||||
self.attn_implementation = attn_implemetation
|
||||
self.attn_implementation = attn_implementation
|
||||
|
||||
def prepare_config_and_inputs(self):
|
||||
input_ids = np.clip(ids_tensor([self.batch_size, self.seq_length - 1], self.vocab_size), 3, self.vocab_size)
|
||||
|
||||
Reference in New Issue
Block a user