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`].
7
7
8
8
After that, simply run the pipeline and save the results.
9
9
10
-
11
10
``` python
12
11
import lighteval
13
12
from lighteval.logging.evaluation_tracker import EvaluationTracker
14
13
from lighteval.models.vllm.vllm_model import VLLMModelConfig
15
14
from lighteval.pipeline import ParallelismManager, Pipeline, PipelineParameters
16
- from lighteval.utils.utils import EnvConfig
17
15
from lighteval.utils.imports import is_accelerate_available
18
16
19
17
if is_accelerate_available():
@@ -33,10 +31,8 @@ def main():
33
31
34
32
pipeline_params = PipelineParameters(
35
33
launcher_type = ParallelismManager.ACCELERATE ,
36
- env_config = EnvConfig(cache_dir = " tmp/" ),
37
34
custom_task_directory = None , # if using a custom task
38
35
# Remove the 2 parameters below once your configuration is tested
39
- override_batch_size = 1 ,
40
36
max_samples = 10
41
37
)
42
38
You can’t perform that action at this time.
0 commit comments