From 2d30bd21558b7f84a4d8529b1b7b856caf8a2190 Mon Sep 17 00:00:00 2001 From: moosecrap Date: Mon, 27 Jul 2026 04:47:54 -0700 Subject: [PATCH] Model switch fix --- tools/generate_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/generate_image.py b/tools/generate_image.py index 5045fee..04fbb0f 100644 --- a/tools/generate_image.py +++ b/tools/generate_image.py @@ -32,7 +32,7 @@ async def ensure_model_state(model_name: str, model_preset: Dict[str, Any]): preset = model_preset.get("preset", "xl") await asyncio.to_thread( requests.post, - f"{config.SD_URL}/api/predict/checkpoint_change", + f"{config.SD_URL}/api/checkpoint_change", json={"data": [target_ckpt, preset]}, timeout=30 ) @@ -45,7 +45,7 @@ async def ensure_model_state(model_name: str, model_preset: Dict[str, Any]): preset = model_preset.get("preset", "xl") await asyncio.to_thread( requests.post, - f"{config.SD_URL}/api/predict/modules_change", + f"{config.SD_URL}/api/modules_change", json={"data": [target_modules, preset]}, timeout=30 )