fix ansi controle output and add model locking

Signed-off-by: Vladimir Mandic <mandic00@live.com>

modules/api/api.py

@@ -94,6 +94,7 @@ class Api:
         self.add_api_route("/sdapi/v1/refresh-checkpoints", endpoints.post_refresh_checkpoints, methods=["POST"])
         self.add_api_route("/sdapi/v1/unload-checkpoint", endpoints.post_unload_checkpoint, methods=["POST"])
         self.add_api_route("/sdapi/v1/reload-checkpoint", endpoints.post_reload_checkpoint, methods=["POST"])
+        self.add_api_route("/sdapi/v1/lock-checkpoint", endpoints.post_lock_checkpoint, methods=["POST"])
         self.add_api_route("/sdapi/v1/refresh-vae", endpoints.post_refresh_vae, methods=["POST"])
         self.add_api_route("/sdapi/v1/latents", endpoints.get_latent_history, methods=["GET"], response_model=List[str])
         self.add_api_route("/sdapi/v1/latents", endpoints.post_latent_history, methods=["POST"], response_model=int)

modules/api/endpoints.py

@@ -115,6 +115,11 @@ def post_reload_checkpoint(force:bool=False):
     sd_models.reload_model_weights()
     return {}
 
+def post_lock_checkpoint(lock:bool=False):
+    from modules import modeldata
+    modeldata.model_data.locked = lock
+    return {}
+
 def get_checkpoint():
     if not shared.sd_loaded or shared.sd_model is None:
         checkpoint = {