Fix failing tests on main due to torch 2.1 (#26607)

* fix * fix * fix --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2023-10-05 10:27:05 +02:00
parent 2ab76c2c4f
commit 54e17a15dc
4 changed files with 10 additions and 0 deletions
--- a/tests/models/wav2vec2_conformer/test_modeling_wav2vec2_conformer.py
+++ b/tests/models/wav2vec2_conformer/test_modeling_wav2vec2_conformer.py
@@ -580,6 +580,7 @@ class Wav2Vec2ConformerModelTest(ModelTesterMixin, PipelineTesterMixin, unittest
            for name, param in model.named_parameters():
                uniform_init_parms = [
                    "conv.weight",
+                    "conv.parametrizations.weight",
                    "masked_spec_embed",
                    "codevectors",
                    "quantizer.weight_proj.weight",