Fixed import issues.

huggingface · Dec 12, 2024 · 932fd2c · 932fd2c
1 parent 7c0856e
commit 932fd2c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/lighteval/main_endpoint.py b/src/lighteval/main_endpoint.py
@@ -26,6 +26,8 @@
 from typer import Argument, Option
 from typing_extensions import Annotated
 
+from lighteval.models.litellm_model import LiteLLMModelConfig
+
 
 app = typer.Typer()
 
@@ -432,7 +434,6 @@ def litellm(
     """
 
     from lighteval.logging.evaluation_tracker import EvaluationTracker
-    from lighteval.models.model_config import LiteLLMModelConfig
     from lighteval.pipeline import EnvConfig, ParallelismManager, Pipeline, PipelineParameters
 
     env_config = EnvConfig(token=TOKEN, cache_dir=cache_dir)

diff --git a/src/lighteval/models/litellm_model.py b/src/lighteval/models/litellm_model.py
@@ -32,7 +32,7 @@
 
 from lighteval.data import GenerativeTaskDataset
 from lighteval.models.abstract_model import LightevalModel
-from lighteval.models.endpoint_model import ModelInfo
+from lighteval.models.endpoints.endpoint_model import ModelInfo
 from lighteval.models.model_output import (
     GenerativeResponse,
     LoglikelihoodResponse,