We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c45f2fa commit 82a1bbcCopy full SHA for 82a1bbc
1 file changed
vllm_rbln/platform.py
@@ -84,10 +84,13 @@ def use_all_gather(cls) -> bool:
84
def pre_register_and_update(cls,
85
parser: Optional[FlexibleArgumentParser] = None
86
) -> None:
87
-
88
if envs.RBLN_USE_VLLM_MODEL:
89
# patches
90
if envs.VLLM_USE_V1:
+ # FIXME(jiwoo.park):disable timeout for RBLN
91
+ import vllm.v1.executor.multiproc_executor as mp
92
+ mp.EXECUTE_MODEL_TIMEOUT_S = None
93
+
94
import vllm_rbln.v1.attention.layer # noqa
95
else:
96
import vllm_rbln.attention.layer # noqa
0 commit comments