github/privateGPT
1
0
Fork 1
mirror of https://github.com/imartinez/privateGPT.git synced 2025-10-13 10:38:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Feature/sagemaker embedding (#1161)

Browse Source 
* Sagemaker deployed embedding model support

---------

Co-authored-by: Pablo Orgaz <pabloogc@gmail.com>
This commit is contained in:
Iván Martínez
2023-11-05 16:16:49 +01:00
committed by GitHub
parent f29df84301
commit ad512e3c42
9 changed files with 114 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
private_gpt/components/llm/llm_component.py
View File

@@ -36,7 +36,7 @@ class LLMComponent:
from private_gpt.components.llm.custom.sagemaker import SagemakerLLM
self.llm = SagemakerLLM(
endpoint_name=settings.sagemaker.endpoint_name,
endpoint_name=settings.sagemaker.llm_endpoint_name,
)
case "openai":
from llama_index.llms import OpenAI