Merge pull request #19 from octo-models/fix-finetune-config

Remove reference to 'no_image_conditioning' relabeling strategy
octo-models · Dec 22, 2023 · bcaddd0 · bcaddd0
2 parents 641a737 + c2e109b
commit bcaddd0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/configs/finetune_config.py b/scripts/configs/finetune_config.py
@@ -97,7 +97,7 @@ def get_config(config_string="full,multimodal"):
         goal_relabeling_strategy = "uniform"
         keep_image_prob = 1.0
     elif task == "language_conditioned":
-        goal_relabeling_strategy = "no_image_conditioning"
+        goal_relabeling_strategy = None
         keep_image_prob = 0.0
     elif task == "multimodal":
         goal_relabeling_strategy = "uniform"