SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

[FlaxSpeechEncoderDecoderModel] Ensure Input and Output Word Embeddings Are **Not** Tied (#16444)

Browse Source 
* [FlaxSpeechEncoderDecoderModel] Ensure Input and Output Word Embeddings Are **Not** Tied

* rebase
This commit is contained in:
Sanchit Gandhi
2022-03-28 14:14:10 +02:00
committed by GitHub
parent e0ac72b7bd
commit 7ca4633555
3 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/speech_encoder_decoder/test_modeling_flax_speech_encoder_decoder.py
View File

@@ -79,6 +79,7 @@ class FlaxEncoderDecoderMixin:
enc_dec_model = FlaxSpeechEncoderDecoderModel(encoder_decoder_config)
self.assertTrue(enc_dec_model.config.is_encoder_decoder)
self.assertFalse(enc_dec_model.config.tie_word_embeddings)
outputs_encoder_decoder = enc_dec_model(
inputs=inputs,

1
tests/speech_encoder_decoder/test_modeling_speech_encoder_decoder.py
View File

@@ -72,6 +72,7 @@ class EncoderDecoderMixin:
enc_dec_model.eval()
self.assertTrue(enc_dec_model.config.is_encoder_decoder)
self.assertFalse(enc_dec_model.config.tie_word_embeddings)
outputs_encoder_decoder = enc_dec_model(
input_values=input_values,