Upload handler.py

handler.py

@@ -101,11 +101,18 @@ class EndpointHandler:
 
             if _is_messages(item):
                 # Chat template path exists in repo; tokenizer.apply_chat_template will use it if configured
-                input_ids = self.tokenizer.apply_chat_template(
+                result = self.tokenizer.apply_chat_template(
                     item,
                     return_tensors="pt",
                     add_generation_prompt=True,
                 )
+                # Handle both single tensor and tuple/dict returns
+                if isinstance(result, dict):
+                    input_ids = result["input_ids"]
+                elif isinstance(result, tuple):
+                    input_ids = result[0]
+                else:
+                    input_ids = result
             else:
                 if not isinstance(item, str):
                     item = str(item)