biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Feature] Add Azure Blob Storage support for RunAI Model Streamer (#34614)

Browse Source 
Signed-off-by: hasethuraman <hsethuraman@microsoft.com>
This commit is contained in:
Hari
2026-03-15 17:08:21 +05:30
committed by GitHub
parent 143e4dccdf
commit a3e2e250f0
14 changed files with 75 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/transformers_utils/runai_utils.py
View File

@@ -13,7 +13,7 @@ from vllm.utils.import_utils import PlaceholderModule
logger = init_logger(__name__)
SUPPORTED_SCHEMES = ["s3://", "gs://"]
SUPPORTED_SCHEMES = ["s3://", "gs://", "az://"]
try:
from runai_model_streamer import list_safetensors as runai_list_safetensors

6
vllm/transformers_utils/utils.py
View File

@@ -23,8 +23,12 @@ def is_gcs(model_or_path: str) -> bool:
return model_or_path.lower().startswith("gs://")
def is_azure(model_or_path: str) -> bool:
return model_or_path.lower().startswith("az://")
def is_cloud_storage(model_or_path: str) -> bool:
return is_s3(model_or_path) or is_gcs(model_or_path)
return is_s3(model_or_path) or is_gcs(model_or_path) or is_azure(model_or_path)
def without_trust_remote_code(kwargs: dict[str, Any]) -> dict[str, Any]: