diff --git a/tests/models/hubert/test_modeling_hubert.py b/tests/models/hubert/test_modeling_hubert.py index 7b3f578861260e..a91402ea805897 100644 --- a/tests/models/hubert/test_modeling_hubert.py +++ b/tests/models/hubert/test_modeling_hubert.py @@ -419,6 +419,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "quantizer.weight_proj.weight", ] @@ -680,6 +681,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "quantizer.weight_proj.weight", ] diff --git a/tests/models/unispeech/test_modeling_unispeech.py b/tests/models/unispeech/test_modeling_unispeech.py index 70c05bd4a6adc7..a286274828e267 100644 --- a/tests/models/unispeech/test_modeling_unispeech.py +++ b/tests/models/unispeech/test_modeling_unispeech.py @@ -421,6 +421,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "codevectors", "quantizer.weight_proj.weight", diff --git a/tests/models/unispeech_sat/test_modeling_unispeech_sat.py b/tests/models/unispeech_sat/test_modeling_unispeech_sat.py index 6e6eeb33d34cda..79fa54717378bb 100644 --- a/tests/models/unispeech_sat/test_modeling_unispeech_sat.py +++ b/tests/models/unispeech_sat/test_modeling_unispeech_sat.py @@ -471,6 +471,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "codevectors", "quantizer.weight_proj.weight", @@ -682,6 +683,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "codevectors", "quantizer.weight_proj.weight", diff --git a/tests/models/wav2vec2/test_modeling_wav2vec2.py b/tests/models/wav2vec2/test_modeling_wav2vec2.py index ffc3ec7399a997..e0f0811cc358a1 100644 --- a/tests/models/wav2vec2/test_modeling_wav2vec2.py +++ b/tests/models/wav2vec2/test_modeling_wav2vec2.py @@ -625,6 +625,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "codevectors", "quantizer.weight_proj.weight", diff --git a/tests/models/wavlm/test_modeling_wavlm.py b/tests/models/wavlm/test_modeling_wavlm.py index ed1654c4877334..6db04e1841f16a 100644 --- a/tests/models/wavlm/test_modeling_wavlm.py +++ b/tests/models/wavlm/test_modeling_wavlm.py @@ -424,6 +424,7 @@ def test_initialization(self): for name, param in model.named_parameters(): uniform_init_parms = [ "conv.weight", + "conv.parametrizations.weight", "masked_spec_embed", "codevectors", "quantizer.weight_proj.weight",