| """Public defaults for the standalone agentic text-to-image upsampler.""" |
|
|
| from __future__ import annotations |
|
|
| from typing import Any |
|
|
| DEFAULT_OPENAI_ENDPOINT_URL = "https://api.openai.com/v1" |
| DEFAULT_UPSAMPLER_ENDPOINT_URL = DEFAULT_OPENAI_ENDPOINT_URL |
| DEFAULT_REWRITER_ENDPOINT_URL = DEFAULT_OPENAI_ENDPOINT_URL |
| DEFAULT_UPSAMPLER_MODEL = "gpt-5.5" |
| DEFAULT_REWRITER_MODEL = "gpt-5.5" |
| DEFAULT_OPENAI_API_KEY_ENV = "OPENAI_API_KEY" |
| DEFAULT_LLM_EXTRA_BODY: dict[str, Any] = {"reasoning_effort": "low"} |
|
|
| DEFAULT_CRITIC_ENDPOINT_URL = "https://generativelanguage.googleapis.com/v1beta/openai/" |
| DEFAULT_CRITIC_MODEL = "gemini-3.1-pro-preview" |
| DEFAULT_GEMINI_API_KEY_ENV = "GEMINI_API_KEY" |
|
|
| DEFAULT_GENERATION_AUTH_KEY_ENV = "AGENTIC_UPSAMPLING_GENERATION_AUTH_KEY" |
| DEFAULT_GENERATION_MODEL = "nvidia/Cosmos3-Super-Text2Image" |
| DEFAULT_IMAGE_SIZE = "1024x1024" |
| DEFAULT_GENERATION_EXTRA_ARGS: dict[str, Any] = {"guardrails": False, "use_resolution_template": False} |
|
|
| DEFAULT_RESOLUTION = "768" |
| DEFAULT_ASPECT_RATIO = "1,1" |
| DEFAULT_NUM_STEPS = 50 |
| DEFAULT_GUIDANCE = 4.0 |
| DEFAULT_FLOW_SHIFT = 3.0 |
| DEFAULT_MAX_ITERATIONS = 2 |
| DEFAULT_SAMPLES_PER_ITERATION = 3 |
| DEFAULT_JPEG_QUALITY = 99 |
|
|
| STRICT_OVERALL_THRESHOLD = 9.0 |
| STRICT_PROMPT_THRESHOLD = 9.0 |
|
|
|
|
