Skip to content

Commit 82a1bbc

Browse files
fix(core): timeout error of large models. (#111)
1 parent c45f2fa commit 82a1bbc

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

vllm_rbln/platform.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -84,10 +84,13 @@ def use_all_gather(cls) -> bool:
8484
def pre_register_and_update(cls,
8585
parser: Optional[FlexibleArgumentParser] = None
8686
) -> None:
87-
8887
if envs.RBLN_USE_VLLM_MODEL:
8988
# patches
9089
if envs.VLLM_USE_V1:
90+
# FIXME(jiwoo.park):disable timeout for RBLN
91+
import vllm.v1.executor.multiproc_executor as mp
92+
mp.EXECUTE_MODEL_TIMEOUT_S = None
93+
9194
import vllm_rbln.v1.attention.layer # noqa
9295
else:
9396
import vllm_rbln.attention.layer # noqa

0 commit comments

Comments
 (0)