henrisel/DeltaVR3DModelGeneration
1
0
Fork 0
forked from cgvr/DeltaVR
Code Pull Requests Activity

invokeAI model key can be set via env var

Browse Source
This commit is contained in:
henrisel
2025-12-19 10:11:18 +02:00
parent b424c6467b
commit f184151fdb
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
3d-generation-pipeline/.env.example
View File

@@ -1,4 +1,5 @@
INVOKEAI_URL= INVOKEAI_URL=
INVOKEAI_MODEL_KEY=
TRELLIS_URL= TRELLIS_URL=
CLOUDFLARE_ACCOUNT_ID= CLOUDFLARE_ACCOUNT_ID=

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

@@ -18,6 +18,7 @@ logger = logging.getLogger("invokeai-mcp")
load_dotenv() load_dotenv()
INVOKEAI_BASE_URL = os.environ["INVOKEAI_URL"] INVOKEAI_BASE_URL = os.environ["INVOKEAI_URL"]
INVOKEAI_MODEL_KEY = os.environ["INVOKEAI_MODEL_KEY"]
DEFAULT_QUEUE_ID = "default" DEFAULT_QUEUE_ID = "default"
# HTTP client # HTTP client
@@ -34,13 +35,12 @@ def get_client() -> httpx.AsyncClient:
async def text_to_image_invoke_ai(prompt, output_path): 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 # see available model keys via GET http://INVOKEAI_BASE_URL:9090/api/v2/models/?model_type=main
args = { args = {
"prompt": prompt, "prompt": prompt,
"width": 512, "width": 512,
"height": 512, "height": 512,
"model_key": "79401292-0a6b-428d-b7d7-f1e86caeba2b" # Juggernaut XL v9 "model_key": INVOKEAI_MODEL_KEY
#"model_key": "735f6485-6703-498f-929e-07cf0bbbd179" # Dreamshaper 8
} }
image_url = await generate_image(args) image_url = await generate_image(args)
print("got image url: ", image_url) print("got image url: ", image_url)