muryshev commited on
Commit
b8ff7b6
·
1 Parent(s): e1080e8

Added hf cache

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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"