Skip to content

Commit 68d4c33

Browse files
[Misc] Add S3 environment variables for better support of MinIO. (#16977)
Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
1 parent f961d7f commit 68d4c33

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

vllm/transformers_utils/s3_utils.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,12 @@ def glob(s3=None,
4545
list[str]: List of full S3 paths allowed by the pattern
4646
"""
4747
if s3 is None:
48-
s3 = boto3.client("s3")
48+
s3 = boto3.client(
49+
's3',
50+
aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
51+
aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
52+
endpoint_url=os.getenv("AWS_ENDPOINT_URL"),
53+
region_name=os.getenv("AWS_REGION_NAME"))
4954
if not path.endswith("/"):
5055
path = path + "/"
5156
bucket_name, _, paths = list_files(s3,
@@ -107,7 +112,12 @@ class S3Model:
107112
"""
108113

109114
def __init__(self) -> None:
110-
self.s3 = boto3.client('s3')
115+
self.s3 = boto3.client(
116+
's3',
117+
aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
118+
aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
119+
endpoint_url=os.getenv("AWS_ENDPOINT_URL"),
120+
region_name=os.getenv("AWS_REGION_NAME"))
111121
for sig in (signal.SIGINT, signal.SIGTERM):
112122
existing_handler = signal.getsignal(sig)
113123
signal.signal(sig, self._close_by_signal(existing_handler))

0 commit comments

Comments
 (0)