diff --git a/LLaVA/llava/model/builder.py b/LLaVA/llava/model/builder.py index 5889e67..6c645bf 100644 --- a/LLaVA/llava/model/builder.py +++ b/LLaVA/llava/model/builder.py @@ -23,9 +23,8 @@ import sys current_dir = os.path.dirname(os.path.abspath(__file__)) sys.path.append(current_dir) -sys.path.append(os.path.abspath(os.path.join(current_dir, '..'))) -from model import * +from . import * from constants import DEFAULT_IMAGE_PATCH_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN