(state_dict)
| 334 | else: |
| 335 | |
| 336 | def remove_prefix_module(state_dict): |
| 337 | # remove the first or the second module |
| 338 | return convert_to_new_checkpoint(state_dict) |
| 339 | |
| 340 | def add_prefix_module(state_dict): |
| 341 | if all(map(lambda x: not x.startswith('module.'), state_dict.keys())): |
no test coverage detected