From 8ea0f677ae427f36453ff6be70bea553e8a95e31 Mon Sep 17 00:00:00 2001
From: pks <pks@pks.rocks>
Date: Sun, 30 Nov 2025 23:01:21 +0100
Subject: WIP

---
 finetuning.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'finetuning.py')

diff --git a/finetuning.py b/finetuning.py
index 853eb81..62f506c 100755
--- a/finetuning.py
+++ b/finetuning.py
@@ -32,8 +32,8 @@ def add_chat_text(example, processor):
     return example
 
 
-def collate(batch, processor, max_length):  # FIXME: Support batch_size > 1
-    images  = [i["image"] for i in batch]
+def collate(batch, processor, max_length):
+    images  = [[i["image"]] for i in batch]
     texts = [i["text"]  for i in batch]
     processor_output = processor(
         text=texts,
-- 
cgit v1.2.3