Convert a single key using compiled rules
(key)
| 526 | rules = get_key_mapping_rules(args.direction, args.model_type) |
| 527 | converted_weights = {} |
| 528 | logger.info("Converting keys...") |
| 529 | |
| 530 | # Pre-compile regex patterns for better performance |
| 531 | compiled_rules = [(re.compile(pattern), replacement) for pattern, replacement in rules] |
| 532 | |
| 533 | def convert_key(key): |
| 534 | """Convert a single key using compiled rules""" |
| 535 | new_key = key |
| 536 | for pattern, replacement in compiled_rules: |