From 728073ef6ae49cfd6b0a3c5618572d810cbb4381 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Thu, 2 Jul 2026 16:14:53 -0400
Subject: [PATCH] fix: use 'output' column (not 'text')

---
 training/configs/ornith-35b-lora.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/training/configs/ornith-35b-lora.yaml b/training/configs/ornith-35b-lora.yaml
index 2cfb555..17c6fab 100644
--- a/training/configs/ornith-35b-lora.yaml
+++ b/training/configs/ornith-35b-lora.yaml
@@ -26,7 +26,7 @@ lora_task_type: CAUSAL_LM
 dataset:
   - path: /home/cyaren/loras/agenx-lora-training/dataset/combined_20k.jsonl
     type: completion
-    text_column: text
+    text_column: output
 
 # Training Parameters
 train_params: