Merge branch 'staging' into SN1-195-add-inference-task

macrocosm-os · Sep 4, 2024 · df9c5fb · df9c5fb
2 parents 117a688 + 4a5cd1a
commit df9c5fb
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/prompting/tasks/task_registry.py b/prompting/tasks/task_registry.py
@@ -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
         ),
     ]