From 41c622c734ef5b36b7aab258fb0851f16d7c2028 Mon Sep 17 00:00:00 2001
From: 51616 <rujikorn.ch@gmail.com>
Date: Wed, 29 Jan 2025 16:48:49 +0000
Subject: [PATCH] fix eval

---
 src/ctx_to_lora/eval.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/ctx_to_lora/eval.py b/src/ctx_to_lora/eval.py
index b294b89..24a4ab9 100644
--- a/src/ctx_to_lora/eval.py
+++ b/src/ctx_to_lora/eval.py
@@ -370,6 +370,7 @@ def evaluate(checkpoint_path, args, split, generative):
         add_repeat_prompt=False,
         add_negative_prompt=False,
         use_kl_loss=False,
+        set_format="pt",
     )
 
     datasets = dict()