From 77d6ccf12b0885773eedd8e4badb244c7ae89b69 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Thu, 30 Nov 2023 12:00:32 -0800 Subject: [PATCH] Add a LOADER debug message while loading models --- modules/models.py | 1 + 1 file changed, 1 insertion(+) diff --git a/modules/models.py b/modules/models.py index 19c0d903..c7dd6ccb 100644 --- a/modules/models.py +++ b/modules/models.py @@ -102,6 +102,7 @@ def load_model(model_name, loader=None): elif loader in ['llama.cpp', 'llamacpp_HF', 'ctransformers']: shared.settings['truncation_length'] = shared.args.n_ctx + logger.info(f"LOADER: {loader}") logger.info(f"TRUNCATION LENGTH: {shared.settings['truncation_length']}") logger.info(f"INSTRUCTION TEMPLATE: {shared.settings['instruction_template']}") logger.info(f"Loaded the model in {(time.time()-t0):.2f} seconds.")