Fix bug

IBM · Feb 18, 2025 · 73a29fb · 73a29fb
1 parent 5af6401
commit 73a29fb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/unitxt/inference.py b/src/unitxt/inference.py
@@ -1156,14 +1156,14 @@ def _infer(
         import ollama
 
         args = self.to_dict([StandardAPIParamsMixin])
-        filtered_args = {k: v for k, v in args.items() if v}
         results = []
-
+        model = args.pop("model")
         for instance in dataset:
             messages = self.to_messages(instance)
             response = ollama.chat(
                 messages=messages,
-                **filtered_args,
+                model=model,
+                options=args,
             )
             results.append(response)