From 90983e8bba77386725687977d8b9ce75f3e4e8c7 Mon Sep 17 00:00:00 2001 From: Heiru Wu Date: Thu, 15 Feb 2024 08:38:11 +0800 Subject: [PATCH] fix(ray): update default resources allocation for test --- instill/helpers/ray_config.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/instill/helpers/ray_config.py b/instill/helpers/ray_config.py index 4daf315..4d1b670 100644 --- a/instill/helpers/ray_config.py +++ b/instill/helpers/ray_config.py @@ -33,10 +33,10 @@ def __init__( # params self.model_weight_or_folder_name: str = model_weight_or_folder_name if use_gpu: - self._update_num_cpus(1) - self._update_num_gpus(0.25) + self._update_num_cpus(0.25) + self._update_num_gpus(0.2) else: - self._update_num_cpus(2) + self._update_num_cpus(0.25) def _update_num_cpus(self, num_cpus: float): if self._deployment.ray_actor_options is not None: