selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-21 04:39:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

add nltk lookup path

Browse Source
This commit is contained in:
Gokul Mohanarangan
2023-09-13 14:13:31 +05:30
parent 9c9bbced47
commit 9a10eef789
2 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
server/reflector/llm/base.py
View File

@@ -51,6 +51,7 @@ class LLM:
nltk.download("punkt", download_dir=settings.CACHE_DIR) nltk.download("punkt", download_dir=settings.CACHE_DIR)
# For POS tagging # For POS tagging
nltk.download("averaged_perceptron_tagger", download_dir=settings.CACHE_DIR) nltk.download("averaged_perceptron_tagger", download_dir=settings.CACHE_DIR)
nltk.data.path.append(settings.CACHE_DIR)
@classmethod @classmethod
def register(cls, name, klass): def register(cls, name, klass):

2
server/reflector/settings.py
View File

@@ -95,7 +95,7 @@ class Settings(BaseSettings):
DEFAULT_LLM: str = "lmsys/vicuna-13b-v1.5" DEFAULT_LLM: str = "lmsys/vicuna-13b-v1.5"
# Cache directory for all model storage # Cache directory for all model storage
CACHE_DIR: str = "data" CACHE_DIR: str = "./data"
settings = Settings() settings = Settings()