From 2d3903db6873f418a48476cba3b0b226be1970fa Mon Sep 17 00:00:00 2001
From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com>
Date: Thu, 26 Sep 2024 09:28:04 +0100
Subject: [PATCH] fix for null checkpoint with API (#1916)

#1915 API oversight
---
 modules/ui_settings.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/ui_settings.py b/modules/ui_settings.py
index be61e5be..112ab9d8 100644
--- a/modules/ui_settings.py
+++ b/modules/ui_settings.py
@@ -333,7 +333,7 @@ class UiSettings:
         button_set_checkpoint = gr.Button('Change checkpoint', elem_id='change_checkpoint', visible=False)
         button_set_checkpoint.click(
             fn=button_set_checkpoint_change,
-            js="function(c, v, n){ var ckpt = desiredCheckpointName; var vae = desiredVAEName; if (vae == 0) vae = v; desiredCheckpointName = null; desiredVAEName = 0; return [ckpt, vae, null]; }",
+            js="function(c, v, n){ var ckpt = desiredCheckpointName; var vae = desiredVAEName; if (ckpt == null) ckpt = c; if (vae == 0) vae = v; desiredCheckpointName = null; desiredVAEName = 0; return [ckpt, vae, null]; }",
             inputs=[main_entry.ui_checkpoint, main_entry.ui_vae, self.dummy_component],
             outputs=[main_entry.ui_checkpoint, main_entry.ui_vae, self.text_settings],
         )