[Feature]: To be compatible with the latest version of MiniGPT-4 (#539)

* [Feature]: To be compatible with the latest version of MiniGPT-4 * [Feature]: User try and except Co-authored-by: Fengzhe Zhou <[email protected]> * [Fix]: Fix lint --------- Co-authored-by: bensenliu <[email protected]> Co-authored-by: Fengzhe Zhou <[email protected]>
open-compass · Nov 4, 2023 · 6e31520 · 6e31520
1 parent f25a980
commit 6e31520
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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)