From 92af8c717642e73379d4ffa751898d682a7fcdf9 Mon Sep 17 00:00:00 2001
From: inkcherry <mingzhi.liu@intel.com>
Date: Mon, 1 Jul 2024 03:26:26 +0000
Subject: [PATCH] automatically added to python module search path

---
 llava/train/train_mem.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/llava/train/train_mem.py b/llava/train/train_mem.py
index 29ea06170..b403b621b 100644
--- a/llava/train/train_mem.py
+++ b/llava/train/train_mem.py
@@ -1,3 +1,12 @@
+import sys
+import os
+
+# Ensure the project root directory is 
+# automatically added to Python module search path
+root_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
+if root_dir not in sys.path:
+    sys.path.insert(0, root_dir)
+    
 from llava.train.train import train
 
 if __name__ == "__main__":