forgot one import

huggingface · Jul 23, 2024 · 96785fd · 96785fd
1 parent e0bd17b
commit 96785fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lighteval/models/model_config.py b/src/lighteval/models/model_config.py
@@ -30,11 +30,11 @@
 
 from lighteval.logging.hierarchical_logger import hlog
 from lighteval.models.utils import _get_model_sha
-from lighteval.pipeline import EnvConfig
 from lighteval.utils import (
     NO_AUTOGPTQ_ERROR_MSG,
     NO_BNB_ERROR_MSG,
     NO_PEFT_ERROR_MSG,
+    EnvConfig,
     is_accelerate_available,
     is_autogptq_available,
     is_bnb_available,