[Wav2Vec2 and Co] Update init tests for PT 2.1 (#26494)

2023-10-03 09:52:34 +01:00
parent b5ca8fcd20
commit 768aa3d9cd
5 changed files with 7 additions and 0 deletions
--- a/tests/models/wavlm/test_modeling_wavlm.py
+++ b/tests/models/wavlm/test_modeling_wavlm.py
@@ -424,6 +424,7 @@ class WavLMModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.TestCase):
            for name, param in model.named_parameters():
                uniform_init_parms = [
                    "conv.weight",
+                    "conv.parametrizations.weight",
                    "masked_spec_embed",
                    "codevectors",
                    "quantizer.weight_proj.weight",