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