Skip to content

Commit ebb9281

Browse files
authored
fix:change multichoice_continuations_start_space category (#610)
move multichoice_continuations_start_space from generation parameter to base_params
1 parent ab14874 commit ebb9281

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/model_configs/transformers_model.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@ model:
33
model_args: "pretrained=HuggingFaceTB/SmolLM-1.7B,revision=main" # pretrained=model_name,trust_remote_code=boolean,revision=revision_to_use,model_parallel=True ...
44
dtype: "bfloat16"
55
compile: true
6+
multichoice_continuations_start_space: null # If true/false, will force multiple choice continuations to start/not start with a space. If none, will do nothing
67
merged_weights: # Ignore this section if you are not using PEFT models
78
delta_weights: false # set to True of your model should be merged with a base model, also need to provide the base model name
89
adapter_weights: false # set to True of your model has been trained with peft, also need to provide the base model name
910
base_model: null # path to the base_model
1011
generation:
11-
multichoice_continuations_start_space: null # If true/false, will force multiple choice continuations to start/not start with a space. If none, will do nothing
1212
temperature: 0.5

src/lighteval/main_accelerate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -168,7 +168,7 @@ def accelerate( # noqa C901
168168
args_dict["accelerator"] = accelerator
169169
args_dict["quantization_config"] = quantization_config
170170
args_dict["batch_size"] = override_batch_size
171-
args_dict["multichoice_continuations_start_space"] = config["generation"][
171+
args_dict["multichoice_continuations_start_space"] = config["base_params"][
172172
"multichoice_continuations_start_space"
173173
]
174174
args_dict["use_chat_template"] = use_chat_template

0 commit comments

Comments
 (0)