From 9a10eef789c11c8046611187407901b90b0955ba Mon Sep 17 00:00:00 2001 From: Gokul Mohanarangan Date: Wed, 13 Sep 2023 14:13:31 +0530 Subject: [PATCH] add nltk lookup path --- server/reflector/llm/base.py | 1 + server/reflector/settings.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/server/reflector/llm/base.py b/server/reflector/llm/base.py index 950a1a07..c93d6099 100644 --- a/server/reflector/llm/base.py +++ b/server/reflector/llm/base.py @@ -51,6 +51,7 @@ class LLM: nltk.download("punkt", download_dir=settings.CACHE_DIR) # For POS tagging nltk.download("averaged_perceptron_tagger", download_dir=settings.CACHE_DIR) + nltk.data.path.append(settings.CACHE_DIR) @classmethod def register(cls, name, klass): diff --git a/server/reflector/settings.py b/server/reflector/settings.py index 3fc45819..249661db 100644 --- a/server/reflector/settings.py +++ b/server/reflector/settings.py @@ -95,7 +95,7 @@ class Settings(BaseSettings): DEFAULT_LLM: str = "lmsys/vicuna-13b-v1.5" # Cache directory for all model storage - CACHE_DIR: str = "data" + CACHE_DIR: str = "./data" settings = Settings()