From 7f02dccc6b02378de69248f8aa435ed1814e2846 Mon Sep 17 00:00:00 2001 From: henrisel Date: Fri, 21 Nov 2025 12:57:53 +0200 Subject: [PATCH] use SD-XL model for img gen in pipeline --- 3d-generation-pipeline/generate_image_local.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/3d-generation-pipeline/generate_image_local.py b/3d-generation-pipeline/generate_image_local.py index de8a5193..c32f52ac 100644 --- a/3d-generation-pipeline/generate_image_local.py +++ b/3d-generation-pipeline/generate_image_local.py @@ -73,8 +73,11 @@ def download_file(url, filepath): async def text_to_image_invoke_ai(prompt, output_path): + # see available model keys via GET http://127.0.0.1:9090/api/v2/models/?model_type=main args = { - "prompt": prompt + "prompt": prompt, + "model_key": "79401292-0a6b-428d-b7d7-f1e86caeba2b" # Juggernaut XL v9 + #"model_key": "735f6485-6703-498f-929e-07cf0bbbd179" # Dreamshaper 8 } image_url = await generate_image(args) print("got image url: ", image_url)