From 6b47f79ec8c5cdb5fe13800c72bb1d069328ba73 Mon Sep 17 00:00:00 2001
From: Fanyi Pu <FPU001@e.ntu.edu.sg>
Date: Fri, 8 Mar 2024 22:44:19 +0800
Subject: [PATCH] update import

---
 lmms_eval/models/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lmms_eval/models/__init__.py b/lmms_eval/models/__init__.py
index 76e82b6e8..d39ce73cf 100644
--- a/lmms_eval/models/__init__.py
+++ b/lmms_eval/models/__init__.py
@@ -11,7 +11,7 @@
 
 for model_name, model_class in AVAILABLE_MODELS.items():
     try:
-        globals()[model_class] = getattr(__import__(model_name, fromlist=[model_class]), model_class)
+        exec(f"from .{model_name} import {model_class}")
     except ImportError:
         pass