Fix typo

Author: mschoenb97

configs/my_config.yaml

@@ -5,4 +5,4 @@ dataset:
     data_path: 'data'
 blip_v2_model_type: blip2-flan-t5-xxl  # Change to blip2-flan-t5-xl for smaller GPUs
 blip_half_precision: True
-# Add more changes here, following the same format as base_config.yaml
+# Add more changes here, following the same format as base_config.yaml

data/queries.csv

@@ -1,2 +1,3 @@
-index,sample_id,possible_answers,query_type,info_to_prompt,query,answer,image_name,img,
-0,0,How many cookies are there?,seven,,How many cookies are there?,seven,cookies.png,cookies.png
+index,sample_id,possible_answers,query_type,query,answer,image_name
+0,0,purple,,What color do you get if you combine the colors of the viper and the flower?,purple,viper_flower.png
+0,0,,,Tell me about the competition between the two skyscrapers in the image.,,skyscrapers.png

main_batch.py

@@ -167,7 +167,6 @@ def main():
                 # TODO compute Codex for next batch as current batch is being processed
 
                 if not config.use_cached_codex:
-                    # codes = codex(prompt=batch['info_to_qprompt'], base_prompt=base_prompt)
                     codes = codex(prompt=batch['query'], base_prompt=base_prompt)
 
                 else:

vision_models.py

@@ -1219,7 +1219,6 @@ def __init__(self, gpu_number=0,
         }
         with warnings.catch_warnings(), HiddenPrints("XVLM"):
             model = XVLMBase(config_xvlm, use_contrastive_loss=True, vision_config=vision_config)
-
             checkpoint = torch.load(path_checkpoint, map_location='cpu')
             state_dict = checkpoint['model'] if 'model' in checkpoint.keys() else checkpoint
             msg = model.load_state_dict(state_dict, strict=False)