github/max_headroom
1
0
Fork 0
mirror of https://github.com/maglore9900/max_headroom.git synced 2025-06-06 03:25:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update adapter.py

Browse Source
This commit is contained in:
maglore9900 2024-10-03 15:48:24 -04:00
parent 9c9878abd9
commit 7cd7a79e7d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/adapter.py
View File

@ -28,7 +28,7 @@ class Adapter:
"answer the following request: {topic}" "answer the following request: {topic}"
) )
self.llm_chat = ChatOllama( self.llm_chat = ChatOllama(
base_url="http://10.0.0.231:11434", model=llm_model base_url=env("OLLAMA_URL"), model=llm_model
) )
model_name = "BAAI/bge-small-en" model_name = "BAAI/bge-small-en"
model_kwargs = {"device": "cpu"} model_kwargs = {"device": "cpu"}