Spaces:
Runtime error
Runtime error
Added hf cache
Browse files
app.py
CHANGED
@@ -16,7 +16,6 @@ llm = LlmBackend()
|
|
16 |
_lock = threading.Lock()
|
17 |
|
18 |
SYSTEM_PROMPT = os.environ.get('SYSTEM_PROMPT') or "Ты — русскоязычный автоматический ассистент. Ты максимально точно и отвечаешь на запросы пользователя, используя русский язык."
|
19 |
-
|
20 |
CONTEXT_SIZE = os.environ.get('CONTEXT_SIZE') or 500
|
21 |
ENABLE_GPU = os.environ.get('ENABLE_GPU') or False
|
22 |
GPU_LAYERS = os.environ.get('GPU_LAYERS') or 0
|
@@ -52,7 +51,7 @@ if os.path.isdir('/data'):
|
|
52 |
|
53 |
model = None
|
54 |
|
55 |
-
MODEL_PATH = snapshot_download(repo_id=repo_name, allow_patterns=model_name) + '/' + model_name
|
56 |
app.logger.info('Model path: ' + MODEL_PATH)
|
57 |
|
58 |
DATASET_REPO_URL = "https://huggingface.co/datasets/muryshev/saiga-chat"
|
|
|
16 |
_lock = threading.Lock()
|
17 |
|
18 |
SYSTEM_PROMPT = os.environ.get('SYSTEM_PROMPT') or "Ты — русскоязычный автоматический ассистент. Ты максимально точно и отвечаешь на запросы пользователя, используя русский язык."
|
|
|
19 |
CONTEXT_SIZE = os.environ.get('CONTEXT_SIZE') or 500
|
20 |
ENABLE_GPU = os.environ.get('ENABLE_GPU') or False
|
21 |
GPU_LAYERS = os.environ.get('GPU_LAYERS') or 0
|
|
|
51 |
|
52 |
model = None
|
53 |
|
54 |
+
MODEL_PATH = snapshot_download(repo_id=repo_name, allow_patterns=model_name, cache_dir='/data/hf_cache') + '/' + model_name
|
55 |
app.logger.info('Model path: ' + MODEL_PATH)
|
56 |
|
57 |
DATASET_REPO_URL = "https://huggingface.co/datasets/muryshev/saiga-chat"
|