Skip to content

Commit

Permalink
Set temperature to 0 (#74)
Browse files Browse the repository at this point in the history
Temperature is 0
  • Loading branch information
eyurtsev authored Mar 21, 2024
1 parent b5b9e5f commit 9240d47
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion backend/server/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,15 @@ def get_supported_models():
)
if "FIREWORKS_API_KEY" in os.environ:
models["fireworks"] = ChatFireworks(
model="accounts/fireworks/models/firefunction-v1"
model="accounts/fireworks/models/firefunction-v1",
temperature=0,
)
if "TOGETHER_API_KEY" in os.environ:
models["together-ai-mistral-8x7b-instruct-v0.1"] = ChatOpenAI(
base_url="https://api.together.xyz/v1",
api_key=os.environ["TOGETHER_API_KEY"],
model="mistralai/Mixtral-8x7B-Instruct-v0.1",
temperature=0,
)

return models
Expand Down

0 comments on commit 9240d47

Please sign in to comment.