diff --git a/src/khoj/migrations/migrate_offline_model.py b/src/khoj/migrations/migrate_offline_model.py index 7ab813ce..15847cbb 100644 --- a/src/khoj/migrations/migrate_offline_model.py +++ b/src/khoj/migrations/migrate_offline_model.py @@ -1,13 +1,18 @@ import os +import logging from khoj.utils.yaml import load_config_from_file, save_config_to_file +logger = logging.getLogger(__name__) + def migrate_offline_model(args): raw_config = load_config_from_file(args.config_file) version = raw_config.get("version") 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. offline_model_path = os.path.expanduser("~/.cache/gpt4all/llama-2-7b-chat.ggmlv3.q4_K_S.bin") if os.path.exists(offline_model_path):