We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 1b718d6 commit 84b3a16Copy full SHA for 84b3a16
vllm/model_executor/model_loader/weight_utils.py
@@ -29,7 +29,9 @@
29
30
try:
31
from runai_model_streamer import SafetensorsStreamer
32
-except ImportError:
+except (ImportError, OSError):
33
+ # see https://github.com/run-ai/runai-model-streamer/issues/26
34
+ # OSError will be raised on arm64 platform
35
runai_model_streamer = PlaceholderModule(
36
"runai_model_streamer") # type: ignore[assignment]
37
SafetensorsStreamer = runai_model_streamer.placeholder_attr(
0 commit comments