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

Add a logline when the offline model migration script runs

Browse Source
This commit is contained in:
sabaimran
2023-08-01 09:27:42 -07:00
parent 48363ec861
commit 778df6be71
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/khoj/migrations/migrate_offline_model.py
View File

@@ -1,13 +1,18 @@
import os import os
import logging
from khoj.utils.yaml import load_config_from_file, save_config_to_file from khoj.utils.yaml import load_config_from_file, save_config_to_file
logger = logging.getLogger(__name__)
def migrate_offline_model(args): def migrate_offline_model(args):
raw_config = load_config_from_file(args.config_file) raw_config = load_config_from_file(args.config_file)
version = raw_config.get("version") version = raw_config.get("version")
if version == "0.10.0" or version == None: if version == "0.10.0" or version == None:
logger.info(f"Migrating offline model used for version {version} to latest version for {args.version_no}")
# If the user has downloaded the offline model, remove it from the cache. # If the user has downloaded the offline model, remove it from the cache.
offline_model_path = os.path.expanduser("~/.cache/gpt4all/llama-2-7b-chat.ggmlv3.q4_K_S.bin") offline_model_path = os.path.expanduser("~/.cache/gpt4all/llama-2-7b-chat.ggmlv3.q4_K_S.bin")
if os.path.exists(offline_model_path): if os.path.exists(offline_model_path):