Skip to content

Commit 78b2992

Browse files
authored
[CI] fix wav2vec test (#42810)
fix ci
1 parent f8e8ddb commit 78b2992

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/transformers/conversion_mapping.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,11 +142,11 @@ def _build_checkpoint_conversion_mapping():
142142
if hasattr(torch.nn.utils.parametrizations, "weight_norm"):
143143
mapping["legacy"] += [
144144
WeightRenaming(
145-
source_patterns=r"weight_g$",
145+
source_patterns="weight_g",
146146
target_patterns="parametrizations.weight.original0",
147147
),
148148
WeightRenaming(
149-
source_patterns=r"weight_v$",
149+
source_patterns="weight_v",
150150
target_patterns="parametrizations.weight.original1",
151151
),
152152
]

0 commit comments

Comments
 (0)