[Feature]: User try and except

Co-authored-by: Fengzhe Zhou <[email protected]>
open-compass · Nov 4, 2023 · b5da708 · b5da708
1 parent ddd684c
commit b5da708
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/opencompass/multimodal/models/minigpt_4/minigpt_4.py b/opencompass/multimodal/models/minigpt_4/minigpt_4.py
@@ -27,10 +27,12 @@ def load_package():
     current_folder_path = os.path.dirname(current_file_path)
 
     sys.path.append(os.path.join(current_folder_path, 'MiniGPT-4'))  # noqa
-    # to be compatible with the latest version
-    # if you are using the old version, please use the following line
-    # 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
+    rom minigpt4.models.mini_gpt4 import MiniGPT4
 
     sys.path.pop(-1)