fix: use correct path to downloaded checkpoint (#88)

blib-la · Mar 19, 2024 · 1071844 · 1071844
1 parent 16b89d6
commit 1071844
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/src/electron/future/ipc/sdk.ts b/src/electron/future/ipc/sdk.ts
@@ -69,7 +69,9 @@ ipcMain.on(
 		const scriptPath = getDirectory("python/live-painting/main.py");
 		const scriptArguments = [
 			"--model_path",
-			getCaptainDownloads("stable-diffusion/checkpoints/stabilityai/sd-turb/fp16"),
+			getCaptainDownloads(
+				"stable-diffusion/checkpoints/stabilityai/sd-turbo/fp16/sd-turbo-fp16"
+			),
 			"--vae_path",
 			getCaptainDownloads("stable-diffusion/vae/madebyollin/taesd"),
 			"--input_image_path",