diff --git a/resources/python/live-painting/main.py b/resources/python/live-painting/main.py index 989ca928f..4a7326822 100644 --- a/resources/python/live-painting/main.py +++ b/resources/python/live-painting/main.py @@ -234,7 +234,7 @@ def warmup(pipe, input_image_path): def main(pipe, input_image_path, output_image_path, shutdown_event): # Initial/default values for parameters - prompt = "realistic face, 4k" + prompt = "a person, incredible watercolor paining" seed = 1 strength = 0.99 guidance_scale = None diff --git a/src/electron/future/ipc/listeners.ts b/src/electron/future/ipc/listeners.ts index 3775250f8..d96a765aa 100644 --- a/src/electron/future/ipc/listeners.ts +++ b/src/electron/future/ipc/listeners.ts @@ -8,7 +8,12 @@ import { execa } from "execa"; import { buildKey } from "#/build-key"; import { DownloadState, ID } from "#/enums"; import { appSettingsStore, userStore } from "@/stores"; -import { getCaptainData, getCaptainDownloads, getDirectory } from "@/utils/path-helpers"; +import { + getCaptainData, + getCaptainDownloads, + getCaptainTemporary, + getDirectory, +} from "@/utils/path-helpers"; import { unpack } from "@/utils/unpack"; ipcMain.on(buildKey([ID.WINDOW], { suffix: ":close" }), () => { @@ -107,15 +112,15 @@ ipcMain.on(buildKey([ID.LIVE_PAINT], { suffix: ":start" }), () => { const scriptPath = getDirectory("python/live-painting/main.py"); const scriptArguments = [ "--model_path", - getCaptainData("downloads/stable-diffusion/checkpoint/sd-turbo/"), + getCaptainDownloads("stable-diffusion/checkpoints/stabilityai/sd-turbo"), "--vae_path", - getCaptainData("downloads/stable-diffusion/vae/taesd/"), + getCaptainDownloads("stable-diffusion/vae/madebyollin/taesd"), "--input_image_path", - getCaptainData("temp/live-painting/input.png"), + getCaptainTemporary("live-painting/input.png"), "--output_image_path", - getCaptainData("temp/live-painting/output.png"), + getCaptainTemporary("live-painting/output.png"), "--disable_stablefast", - // "--debug", + "--debug", ]; process = execa(pythonBinaryPath, ["-u", scriptPath, ...scriptArguments]); diff --git a/src/electron/future/utils/path-helpers.ts b/src/electron/future/utils/path-helpers.ts index 5ca1941b8..0e7ca23ba 100644 --- a/src/electron/future/utils/path-helpers.ts +++ b/src/electron/future/utils/path-helpers.ts @@ -35,3 +35,7 @@ export function getCaptainData(...subpath: string[]): string { export function getCaptainDownloads(...subpath: string[]): string { return getCaptainData("downloads", ...subpath); } + +export function getCaptainTemporary(...subpath: string[]): string { + return getCaptainData("temp", ...subpath); +}