correct task instruction

guestrin-lab · Nov 15, 2024 · a97225a · a97225a
1 parent a0dc95c
commit a97225a
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/lotus/templates/task_instructions.py b/lotus/templates/task_instructions.py
@@ -35,10 +35,7 @@ def filter_user_message_formatter(
     if not image_inputs or len(image_inputs) == 0:
         return {
             "role": "user",
-            "content": {
-                "type": "text",
-                "text": f"Claim: {user_instruction}\n\nContext:\n{text}",
-            },
+            "content": f"Claim: {user_instruction}\n\nContext:\n{text}",
         }
     return {
         "role": "user",