From 8fbfe952f8bd9882cf501df0ffdb56c609ab5ff0 Mon Sep 17 00:00:00 2001 From: Rene Teigen Date: Tue, 3 Jan 2023 20:03:01 +0000 Subject: [PATCH] Changed how the overrides are checked to allow "0" input --- models/openai_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/openai_model.py b/models/openai_model.py index f6e85bc..96d2d6e 100644 --- a/models/openai_model.py +++ b/models/openai_model.py @@ -380,16 +380,16 @@ class Model: payload = { "model": self.model, "prompt": prompt, - "temperature": self.temp if not temp_override else temp_override, - "top_p": self.top_p if not top_p_override else top_p_override, + "temperature": self.temp if temp_override is None else temp_override, + "top_p": self.top_p if top_p_override is None else top_p_override, "max_tokens": self.max_tokens - tokens if not max_tokens_override else max_tokens_override, "presence_penalty": self.presence_penalty - if not presence_penalty_override + if presence_penalty_override is None else presence_penalty_override, "frequency_penalty": self.frequency_penalty - if not frequency_penalty_override + if frequency_penalty_override is None else frequency_penalty_override, "best_of": self.best_of if not best_of_override else best_of_override, }