cgvr/DeltaVR
1
0
Fork 1
Code Issues 7 Pull Requests Actions Packages Projects Releases Wiki Activity

use SD-XL model for img gen in pipeline

Browse Source
This commit is contained in:
henrisel 2025-11-21 12:57:53 +02:00
parent 34a6c50598
commit 7f02dccc6b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
3d-generation-pipeline/generate_image_local.py
View File

@ -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)