diff --git a/docs/source/using-the-python-api.mdx b/docs/source/using-the-python-api.mdx index 9ad7f3c90..7e696219e 100644 --- a/docs/source/using-the-python-api.mdx +++ b/docs/source/using-the-python-api.mdx @@ -7,13 +7,11 @@ and a [`~pipeline.Pipeline`]. After that, simply run the pipeline and save the results. - ```python import lighteval from lighteval.logging.evaluation_tracker import EvaluationTracker from lighteval.models.vllm.vllm_model import VLLMModelConfig from lighteval.pipeline import ParallelismManager, Pipeline, PipelineParameters -from lighteval.utils.utils import EnvConfig from lighteval.utils.imports import is_accelerate_available if is_accelerate_available(): @@ -33,10 +31,8 @@ def main(): pipeline_params = PipelineParameters( launcher_type=ParallelismManager.ACCELERATE, - env_config=EnvConfig(cache_dir="tmp/"), custom_task_directory=None, # if using a custom task # Remove the 2 parameters below once your configuration is tested - override_batch_size=1, max_samples=10 )