github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

change ak sk name

pull/527/head
lijiaxing 2023-12-05 13:57:05 +08:00
parent 5c2c247e21
commit 3410362f4c
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
internlm/utils/storage_manager.py
View File

@ -328,8 +328,8 @@ class Boto3Client(StorageClient):
"""S3 object/file storage management class
Args:
s3_access_keys_id (str): S3 access key ID.
s3_secret_access_key (str): S3 secret access key.
ACCESS_KEY (str): S3 access key ID.
SECRET_ACCESS_KEY (str): S3 secret access key.
use_threads (bool, optional): Whether to enable multipart. Defaults to True.
multipart_chunksize (_type_, optional): Defaults to 8*MB.
max_concurrency (int, optional): Defaults to 10.
@ -340,11 +340,11 @@ class Boto3Client(StorageClient):
super().__init__(boto3)
self.botocore = botocore
try:
s3_access_key_id = os.environ["S3_ACCESS_KEY_ID"]
s3_secret_access_key = os.environ["S3_SECRET_ACCESS_KEY_ID"]
s3_access_key_id = os.environ["ACCESS_KEY "]
s3_secret_access_key = os.environ["SECRET_ACCESS_KEY "]
except KeyError as exc:
raise RuntimeError(
"Please set boto3 bucket 'S3_ACCESS_KEY_ID' and 'S3_SECRET_ACCESS_KEY_ID' using environment variable!"
"Please set boto3 bucket 'ACCESS_KEY' and 'SECRET_ACCESS_KEY' using environment variable!"
) from exc
self.client = self.handler.client(