Add a LOADER debug message while loading models

oobabooga · Nov 30, 2023 · 77d6ccf · 77d6ccf
1 parent 1c90e02
commit 77d6ccf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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.")