[Feature] Add Azure Blob Storage support for RunAI Model Streamer (#34614)
Signed-off-by: hasethuraman <hsethuraman@microsoft.com>
This commit is contained in:
@@ -11,6 +11,7 @@ from vllm.transformers_utils.gguf_utils import (
|
||||
split_remote_gguf,
|
||||
)
|
||||
from vllm.transformers_utils.utils import (
|
||||
is_azure,
|
||||
is_cloud_storage,
|
||||
is_gcs,
|
||||
is_s3,
|
||||
@@ -31,9 +32,17 @@ def test_is_s3():
|
||||
assert not is_s3("nfs://nfs-fqdn.local")
|
||||
|
||||
|
||||
def test_is_azure():
|
||||
assert is_azure("az://model-container/path")
|
||||
assert not is_azure("s3://model-path/path-to-model")
|
||||
assert not is_azure("/unix/local/path")
|
||||
assert not is_azure("nfs://nfs-fqdn.local")
|
||||
|
||||
|
||||
def test_is_cloud_storage():
|
||||
assert is_cloud_storage("gs://model-path")
|
||||
assert is_cloud_storage("s3://model-path/path-to-model")
|
||||
assert is_cloud_storage("az://model-container/path")
|
||||
assert not is_cloud_storage("/unix/local/path")
|
||||
assert not is_cloud_storage("nfs://nfs-fqdn.local")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user