From ebef3bb3a79fef4bc8aa802ae3488e976e965f8f Mon Sep 17 00:00:00 2001 From: Heiru Wu Date: Tue, 10 Sep 2024 01:09:21 +0800 Subject: [PATCH] fix(ray): fix number of sample type --- instill/helpers/ray_io.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/instill/helpers/ray_io.py b/instill/helpers/ray_io.py index ab98137..2a33bce 100644 --- a/instill/helpers/ray_io.py +++ b/instill/helpers/ray_io.py @@ -441,7 +441,7 @@ async def parse_task_completion_to_completion_input( # number of generated outputs if "n" in parameter: - inp.n = parameter["n"] + inp.n = int(parameter["n"]) # top p if "top-p" in parameter: @@ -640,7 +640,7 @@ async def parse_task_chat_to_chat_input( # number of generated outputs if "n" in parameter: - inp.n = parameter["n"] + inp.n = int(parameter["n"]) # top p if "top-p" in parameter: @@ -830,7 +830,7 @@ async def parse_task_chat_to_multimodal_chat_input( # number of generated outputs if "n" in parameter: - inp.n = parameter["n"] + inp.n = int(parameter["n"]) # top p if "top-p" in parameter: @@ -886,7 +886,7 @@ async def parse_task_text_to_image_input( # number of generated outputs if "n" in parameter: - inp.n = parameter["n"] + inp.n = int(parameter["n"]) # seed if "seed" in parameter: