klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-07 13:23:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Run GPT4All Chat Model on GPU, when available

Browse Source 
GPT4All now supports running models on GPU via Vulkan
This commit is contained in:
Debanjum Singh Solanky
2023-10-04 18:42:12 -07:00
parent 13b16a4364
commit d1ff812021
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/khoj/processor/conversation/gpt4all/utils.py
View File

@@ -11,4 +11,12 @@ def download_model(model_name: str):
logger.info("There was an error importing GPT4All. Please run pip install gpt4all in order to install it.")
raise e
return GPT4All(model_name=model_name)
# Use GPU for Chat Model, if available
try:
model = GPT4All(model_name=model_name, device="gpu")
logger.debug("Loaded chat model to GPU.")
except ValueError:
model = GPT4All(model_name=model_name)
logger.debug("Loaded chat model to CPU.")
return model