phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-12 12:06:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
Swapna Lekkala 2025-10-12 06:48:50 -07:00
parent bc5eeef6f3
commit 1954b60600
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/safety/nvidia/nvidia.py
View file

@ -133,7 +133,7 @@ class NeMoGuardrails:
""" """
request_data = { request_data = {
"model": self.model, "model": self.model,
"messages": messages, "messages": [{"role": message.role, "content": message.content} for message in messages],
"temperature": self.temperature, "temperature": self.temperature,
"top_p": 1, "top_p": 1,
"frequency_penalty": 0, "frequency_penalty": 0,