xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-24 21:05:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Misc] Add S3 environment variables for better support of MinIO. (#16977)

Browse Source 
Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
This commit is contained in:
Chauncey 2025-04-22 22:27:36 +08:00 committed by GitHub
parent f961d7f6ef
commit 68d4c33202
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
vllm/transformers_utils/s3_utils.py
View File

@ -45,7 +45,12 @@ def glob(s3=None,
list[str]: List of full S3 paths allowed by the pattern
"""
if s3 is None:
s3 = boto3.client("s3")
s3 = boto3.client(
's3',
aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
endpoint_url=os.getenv("AWS_ENDPOINT_URL"),
region_name=os.getenv("AWS_REGION_NAME"))
if not path.endswith("/"):
path = path + "/"
bucket_name, _, paths = list_files(s3,
@ -107,7 +112,12 @@ class S3Model:
"""
def __init__(self) -> None:
self.s3 = boto3.client('s3')
self.s3 = boto3.client(
's3',
aws_access_key_id=os.getenv("AWS_ACCESS_KEY_ID"),
aws_secret_access_key=os.getenv("AWS_SECRET_ACCESS_KEY"),
endpoint_url=os.getenv("AWS_ENDPOINT_URL"),
region_name=os.getenv("AWS_REGION_NAME"))
for sig in (signal.SIGINT, signal.SIGTERM):
existing_handler = signal.getsignal(sig)
signal.signal(sig, self._close_by_signal(existing_handler))