omar-ah
/

ViL-DLM-0.6B

Image-Text-to-Text

vision-language

masked-diffusion

Model card Files Files and versions

omar-ah commited on 14 days ago

Commit

f089e8f

·

1 Parent(s): 2dd6eee

Fix The Cauldron aokvqa config name

Files changed (2) hide show

code/model_config.py +1 -1
code/train_production.py +1 -1

code/model_config.py CHANGED Viewed

@@ -101,7 +101,7 @@ class TrainingConfig:
     finetune_dataset_configs: List[str] = field(default_factory=lambda: [
         "ai2d",
         "vqav2",
-        "a_okvqa",
         "textvqa",
         "docvqa",
         "chartqa",

     finetune_dataset_configs: List[str] = field(default_factory=lambda: [
         "ai2d",
         "vqav2",
+        "aokvqa",
         "textvqa",
         "docvqa",
         "chartqa",

code/train_production.py CHANGED Viewed

@@ -49,7 +49,7 @@ from vision_xlstm import (
 DEFAULT_CAULDRON_CONFIGS = [
     "ai2d",
     "vqav2",
-    "a_okvqa",
     "textvqa",
     "docvqa",
     "chartqa",

 DEFAULT_CAULDRON_CONFIGS = [
     "ai2d",
     "vqav2",
+    "aokvqa",
     "textvqa",
     "docvqa",
     "chartqa",