Update adapter.py

This commit is contained in:
maglore9900 2024-10-03 15:48:24 -04:00
parent 9c9878abd9
commit 7cd7a79e7d

View File

@ -28,7 +28,7 @@ class Adapter:
"answer the following request: {topic}"
)
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_kwargs = {"device": "cpu"}