update import

EvolvingLMMs-Lab · Mar 8, 2024 · 6b47f79 · 6b47f79
1 parent 6f657ec
commit 6b47f79
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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