[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/unispeech/test_modeling_unispeech.py
+++ b/tests/models/unispeech/test_modeling_unispeech.py
@@ -421,6 +421,7 @@ class UniSpeechRobustModelTest(ModelTesterMixin, PipelineTesterMixin, unittest.T
            for name, param in model.named_parameters():
                uniform_init_parms = [
                    "conv.weight",
+                    "conv.parametrizations.weight",
                    "masked_spec_embed",
                    "codevectors",
                    "quantizer.weight_proj.weight",