diff --git a/opencompass/multimodal/models/minigpt_4/minigpt_4.py b/opencompass/multimodal/models/minigpt_4/minigpt_4.py index 3c0986d9..f722526c 100644 --- a/opencompass/multimodal/models/minigpt_4/minigpt_4.py +++ b/opencompass/multimodal/models/minigpt_4/minigpt_4.py @@ -27,7 +27,13 @@ def load_package(): current_folder_path = os.path.dirname(current_file_path) sys.path.append(os.path.join(current_folder_path, 'MiniGPT-4')) # noqa - from minigpt4.models.mini_gpt4 import MiniGPT4 + + try: + # the latest version of MiniGPT4 + from minigpt4.models.minigpt4 import MiniGPT4 + except ImportError: + # the old version of MiniGPT4 + from minigpt4.models.mini_gpt4 import MiniGPT4 sys.path.pop(-1)