File tree Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Expand file tree Collapse file tree 1 file changed +0
-4
lines changed Original file line number Diff line number Diff line change @@ -7,13 +7,11 @@ and a [`~pipeline.Pipeline`].
77
88After that, simply run the pipeline and save the results.
99
10-
1110``` python
1211import lighteval
1312from lighteval.logging.evaluation_tracker import EvaluationTracker
1413from lighteval.models.vllm.vllm_model import VLLMModelConfig
1514from lighteval.pipeline import ParallelismManager, Pipeline, PipelineParameters
16- from lighteval.utils.utils import EnvConfig
1715from lighteval.utils.imports import is_accelerate_available
1816
1917if is_accelerate_available():
@@ -33,10 +31,8 @@ def main():
3331
3432 pipeline_params = PipelineParameters(
3533 launcher_type = ParallelismManager.ACCELERATE ,
36- env_config = EnvConfig(cache_dir = " tmp/" ),
3734 custom_task_directory = None , # if using a custom task
3835 # Remove the 2 parameters below once your configuration is tested
39- override_batch_size = 1 ,
4036 max_samples = 10
4137 )
4238
You can’t perform that action at this time.
0 commit comments