diff --git a/utils/modular_model_converter.py b/utils/modular_model_converter.py index 0ae641d4e5..d2f0a99fc9 100644 --- a/utils/modular_model_converter.py +++ b/utils/modular_model_converter.py @@ -1088,8 +1088,9 @@ def convert_modular_file(modular_file, old_model_name=None, new_model_name=None, if node != {}: # Get relative path starting from src/transformers/ relative_path = re.search( - f"{os.sep}(src{os.sep}transformers{os.sep}.*)", os.path.abspath(modular_file) + rf"(src{os.sep}transformers{os.sep}.*|examples{os.sep}.*)", os.path.abspath(modular_file) ).group(1) + header = AUTO_GENERATED_MESSAGE.format( relative_path=relative_path, short_name=os.path.basename(relative_path) )