From e7dd88970fddea62006ba7b6620db6a31c97f5ed Mon Sep 17 00:00:00 2001 From: pks Date: Mon, 1 Dec 2025 12:07:08 +0100 Subject: WIP --- make_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'make_dataset.py') diff --git a/make_dataset.py b/make_dataset.py index f996acc..65338d5 100755 --- a/make_dataset.py +++ b/make_dataset.py @@ -26,7 +26,7 @@ def make_dataset(base="./baseline"): # TODO: Make actual hf dataset }, ensure_ascii=False, indent=0)) images.append(image_path) - return Dataset.from_dict({"image": images, "user": user_prompts, "assistant": assistant_replies}).cast_column("image", Image()) + return Dataset.from_dict({"image": images, "user": user_prompts, "assistant": assistant_replies}).cast_column("image", Image().convert("RGB")) def main(): -- cgit v1.2.3