wip

src/compute_horde_prompt_gen/model.py

@@ -80,7 +80,7 @@ def tokenize(prompt: str) -> str:
     def tokenize_phi3(self, prompts: list[str], role: str) -> str:
         inputs = [{"role": role, "content": prompt} for prompt in prompts]
         inputs = self.tokenizer.apply_chat_template(
-            prompts, add_generation_prompt=True, return_tensors="pt"
+            inputs, add_generation_prompt=True, return_tensors="pt"
         )
         return inputs