Archives/DocsGPT
2
0
Fork 0
mirror of https://github.com/arc53/DocsGPT synced 2024-11-17 21:26:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update sagemaker.py

Browse Source
This commit is contained in:
Alex 2023-10-06 14:43:05 +01:00
parent b3a0368b95
commit 43a22f84d9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
application/llm/sagemaker.py
View File

@ -64,9 +64,9 @@ class SagemakerAPILLM(BaseLLM):
import boto3
runtime = boto3.client(
'runtime.sagemaker',
aws_access_key_id=settings.SAGEMAKER_ACCESS_KEY,
aws_secret_access_key=settings.SAGEMAKER_SECRET_KEY,
region_name=settings.SAGEMAKER_REGION
aws_access_key_id='xxx',
aws_secret_access_key='xxx',
region_name='us-west-2'
)