valhalla commited on
Commit
a265819
1 Parent(s): f6c4cb2

don't tie embeddings

Browse files
Files changed (1) hide show
  1. dalle_mini/configuration_bart.py +2 -0
dalle_mini/configuration_bart.py CHANGED
@@ -139,6 +139,7 @@ class BartConfig(PretrainedConfig):
139
  is_encoder_decoder=True,
140
  decoder_start_token_id=16384,
141
  forced_eos_token_id=2,
 
142
  **kwargs,
143
  ):
144
  self.vocab_size = vocab_size
@@ -173,6 +174,7 @@ class BartConfig(PretrainedConfig):
173
  is_encoder_decoder=is_encoder_decoder,
174
  decoder_start_token_id=decoder_start_token_id,
175
  forced_eos_token_id=forced_eos_token_id,
 
176
  **kwargs,
177
  )
178
 
 
139
  is_encoder_decoder=True,
140
  decoder_start_token_id=16384,
141
  forced_eos_token_id=2,
142
+ tie_word_embeddings=False, # don't tie for scaling reasons
143
  **kwargs,
144
  ):
145
  self.vocab_size = vocab_size
 
174
  is_encoder_decoder=is_encoder_decoder,
175
  decoder_start_token_id=decoder_start_token_id,
176
  forced_eos_token_id=forced_eos_token_id,
177
+ tie_word_embeddings=tie_word_embeddings,
178
  **kwargs,
179
  )
180