Skip to content

Commit 174c499

Browse files
ahao-anyscalertourgeman
authored andcommitted
[BUG] Allow runai_streamer_sharded in config check (vllm-project#26958)
Signed-off-by: ahao-anyscale <ahao@anyscale.com>
1 parent 29d61d5 commit 174c499

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

vllm/config/vllm.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -740,10 +740,13 @@ def try_verify_and_update_config(self):
740740
"Overriding `load_format` to 'runai_streamer'"
741741
)
742742
self.load_config.load_format = "runai_streamer"
743-
elif self.load_config.load_format != "runai_streamer":
743+
elif self.load_config.load_format not in (
744+
"runai_streamer",
745+
"runai_streamer_sharded",
746+
):
744747
raise ValueError(
745748
f"To load a model from S3, 'load_format' "
746-
f"must be 'runai_streamer', "
749+
f"must be 'runai_streamer' or 'runai_streamer_sharded', "
747750
f"but got '{self.load_config.load_format}'. "
748751
f"Model: {self.model_config.model}"
749752
)

0 commit comments

Comments
 (0)