Skip to content

Commit e9000ef

Browse files
committed
fix
1 parent b2b654a commit e9000ef

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/models/seamless_m4t_v2/test_modeling_seamless_m4t_v2.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -76,6 +76,9 @@ def __init__(
7676
decoder_layers=2,
7777
encoder_ffn_dim=6,
7878
decoder_ffn_dim=6,
79+
encoder_layerdrop=0.0,
80+
speech_encoder_layerdrop=0.0,
81+
decoder_layerdrop=0.0,
7982
t2u_encoder_layers=2,
8083
t2u_decoder_layers=2,
8184
t2u_encoder_ffn_dim=6,
@@ -154,6 +157,10 @@ def __init__(
154157
self.speech_encoder_chunk_size = speech_encoder_chunk_size
155158
self.speech_encoder_left_chunk_num = speech_encoder_left_chunk_num
156159

160+
self.encoder_layerdrop = encoder_layerdrop
161+
self.speech_encoder_layerdrop = speech_encoder_layerdrop
162+
self.decoder_layerdrop = decoder_layerdrop
163+
157164
def prepare_config_and_inputs(self):
158165
if self.input_modality == "text":
159166
inputs = ids_tensor([self.batch_size, self.seq_length], self.vocab_size - 1)

0 commit comments

Comments
 (0)