From fe0c391e8fc12b609156b05108acb1d3e8bfa033 Mon Sep 17 00:00:00 2001 From: Henk Date: Wed, 2 Aug 2023 10:50:00 +0200 Subject: [PATCH] Only show stopped if started --- modeling/inference_models/hf_mtj/class.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modeling/inference_models/hf_mtj/class.py b/modeling/inference_models/hf_mtj/class.py index c0f70843..a4600465 100644 --- a/modeling/inference_models/hf_mtj/class.py +++ b/modeling/inference_models/hf_mtj/class.py @@ -133,7 +133,8 @@ class model_backend(HFInferenceModel): utils.koboldai_vars.compiling = True def mtj_stopped_compiling_callback() -> None: - print(Colors.GREEN + "TPU backend compilation stopped" + Colors.END) + if utils.koboldai_vars.compiling: + print(Colors.GREEN + "TPU backend compilation stopped" + Colors.END) utils.koboldai_vars.compiling = False def mtj_settings_callback() -> dict: