Skip to content

Commit

Permalink
Merge branch 'SN1-195-add-inference-task' of https://github.com/macro…
Browse files Browse the repository at this point in the history
…cosm-os/prompting into SN1-195-add-inference-task
  • Loading branch information
Hollyqui committed Sep 4, 2024
2 parents d6350f9 + df9c5fb commit 06e6faa
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions prompting/tasks/task_registry.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,10 +43,7 @@ class TaskRegistry(BaseModel):
reward_model=InferenceRewardConfig,
),
TaskConfig(
task=MultiChoiceTask,
probability=0.1,
datasets=[WikiDataset],
reward_model=MultiChoiceRewardConfig,
task=MultiChoiceTask, probability=0.2, datasets=[WikiDataset], reward_model=MultiChoiceRewardConfig
),
]

Expand Down

0 comments on commit 06e6faa

Please sign in to comment.