From 24b82f3cd56d5eeb26e649207aac3ce8a7d75bdc Mon Sep 17 00:00:00 2001
From: Mohamed Mekkouri <93391238+MekkCyber@users.noreply.github.com>
Date: Thu, 10 Oct 2024 18:43:27 +0200
Subject: [PATCH] Small Fix to modular converter (#34051)

* small_fix

* supporting both src/tranformers and examples/

* make style
---
 utils/modular_model_converter.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

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)
                 )