Update train_vlm.py

train_vlm.py

@@ -42,7 +42,7 @@ def load_and_prepare_dataset(stage, image_processor, tokenizer, model, split="tr
         image = examples['image'].convert("RGB")
         # --- USING THE CONVERSATION FORMAT YOU PROVIDED ---
         # We select 'conversation_a' and parse it as a list of lists of dicts.
-        conversation = examples['conversation_a']
+        conversation = examples['conversation']
         
         full_text = ""
         is_first_user_turn = True