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

pre-commit updates

Browse source
This commit is contained in:
Sajikumar JS 2025-04-26 18:29:47 +05:30
parent 7f1e4bf075
commit a9e4d1f00e
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/remote/inference/watsonx/watsonx.py
View file

@ -29,12 +29,12 @@ from llama_stack.apis.inference import (
ToolPromptFormat,
)
from llama_stack.apis.inference.inference import (
GreedySamplingStrategy,
OpenAIChatCompletion,
OpenAIChatCompletionChunk,
OpenAICompletion,
OpenAIMessageParam,
OpenAIResponseFormatParam,
GreedySamplingStrategy,
TopKSamplingStrategy,
TopPSamplingStrategy,
)
@ -242,8 +242,7 @@ class WatsonXInferenceAdapter(Inference, ModelRegistryHelper):
if isinstance(request.sampling_params.strategy, GreedySamplingStrategy):
input_dict["params"][GenParams.TEMPERATURE] = 0.0
input_dict["params"][GenParams.STOP_SEQUENCES] = ["<|endoftext|>"]
input_dict["params"][GenParams.STOP_SEQUENCES] = ["<|endoftext|>"]
params = {
**input_dict,