Skip to content

Commit 7a76419

Browse files
authored
[bugfix] fix qwen3_omni all_linear aligner (modelscope#8105)
1 parent 2f34cc7 commit 7a76419

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

swift/megatron/utils/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,7 @@ def get_multimodal_target_regex(
9191
target_modules = [tm for tm in target_modules if tm]
9292
target_pattern = rf'.*\.({"|".join(target_modules)})' if target_modules else ''
9393
rejected_pattern = rf'(?!({"|".join(rejected_modules)}))' if rejected_modules else ''
94-
res.append(rf'{rejected_pattern}{module}(?=\.){target_pattern}')
94+
res.append(rf'{rejected_pattern}{re.escape(module)}(?=\.){target_pattern}')
9595

9696
return rf'^({"|".join(res)})$'
9797

swift/utils/transformers_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def get_multimodal_target_regex(
246246
target_modules = [tm for tm in target_modules if tm]
247247
target_pattern = rf'.*\.({"|".join(target_modules)})' if target_modules else ''
248248
rejected_pattern = rf'(?!({"|".join(rejected_modules)}))' if rejected_modules else ''
249-
res.append(rf'{rejected_pattern}{module}{target_pattern}')
249+
res.append(rf'{rejected_pattern}{re.escape(module)}(?=\.){target_pattern}')
250250

251251
return rf'^({"|".join(res)})$'
252252

0 commit comments

Comments
 (0)