Update Modules/Generate_Image.py
Browse files
Modules/Generate_Image.py
CHANGED
|
@@ -34,10 +34,10 @@ def Generate_Image(
|
|
| 34 |
),
|
| 35 |
steps: Annotated[int, "Number of denoising steps (1–100). Higher = slower, potentially higher quality."] = 35,
|
| 36 |
cfg_scale: Annotated[float, "Classifier-free guidance scale (1–20). Higher = follow the prompt more closely."] = 7.0,
|
| 37 |
-
sampler: Annotated[str, "Sampling method label (UI only). Common options: 'DPM++ 2M Karras', 'DPM++ SDE Karras', 'Euler', 'Euler a', 'Heun', 'DDIM'."] = "DPM++ 2M Karras",
|
| 38 |
seed: Annotated[int, "Random seed for reproducibility. Use -1 for a random seed per call."] = -1,
|
| 39 |
width: Annotated[int, "Output width in pixels (64–1216, multiple of 32 recommended)."] = 1024,
|
| 40 |
height: Annotated[int, "Output height in pixels (64–1216, multiple of 32 recommended)."] = 1024,
|
|
|
|
| 41 |
) -> Image.Image:
|
| 42 |
_log_call_start(
|
| 43 |
"Generate_Image",
|
|
@@ -109,14 +109,14 @@ def build_interface() -> gr.Interface:
|
|
| 109 |
),
|
| 110 |
gr.Slider(minimum=1, maximum=100, value=35, step=1, label="Steps"),
|
| 111 |
gr.Slider(minimum=1.0, maximum=20.0, value=7.0, step=0.1, label="CFG Scale"),
|
|
|
|
|
|
|
|
|
|
| 112 |
gr.Radio(
|
| 113 |
label="Sampler",
|
| 114 |
value="DPM++ 2M Karras",
|
| 115 |
choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"],
|
| 116 |
),
|
| 117 |
-
gr.Slider(minimum=-1, maximum=1_000_000_000, value=-1, step=1, label="Seed (-1 = random)"),
|
| 118 |
-
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Width"),
|
| 119 |
-
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Height"),
|
| 120 |
],
|
| 121 |
outputs=gr.Image(label="Generated Image"),
|
| 122 |
title="Generate Image",
|
|
|
|
| 34 |
),
|
| 35 |
steps: Annotated[int, "Number of denoising steps (1–100). Higher = slower, potentially higher quality."] = 35,
|
| 36 |
cfg_scale: Annotated[float, "Classifier-free guidance scale (1–20). Higher = follow the prompt more closely."] = 7.0,
|
|
|
|
| 37 |
seed: Annotated[int, "Random seed for reproducibility. Use -1 for a random seed per call."] = -1,
|
| 38 |
width: Annotated[int, "Output width in pixels (64–1216, multiple of 32 recommended)."] = 1024,
|
| 39 |
height: Annotated[int, "Output height in pixels (64–1216, multiple of 32 recommended)."] = 1024,
|
| 40 |
+
sampler: Annotated[str, "Sampling method label (UI only). Common options: 'DPM++ 2M Karras', 'DPM++ SDE Karras', 'Euler', 'Euler a', 'Heun', 'DDIM'."] = "DPM++ 2M Karras",
|
| 41 |
) -> Image.Image:
|
| 42 |
_log_call_start(
|
| 43 |
"Generate_Image",
|
|
|
|
| 109 |
),
|
| 110 |
gr.Slider(minimum=1, maximum=100, value=35, step=1, label="Steps"),
|
| 111 |
gr.Slider(minimum=1.0, maximum=20.0, value=7.0, step=0.1, label="CFG Scale"),
|
| 112 |
+
gr.Slider(minimum=-1, maximum=1_000_000_000, value=-1, step=1, label="Seed (-1 = random)"),
|
| 113 |
+
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Width"),
|
| 114 |
+
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Height"),
|
| 115 |
gr.Radio(
|
| 116 |
label="Sampler",
|
| 117 |
value="DPM++ 2M Karras",
|
| 118 |
choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"],
|
| 119 |
),
|
|
|
|
|
|
|
|
|
|
| 120 |
],
|
| 121 |
outputs=gr.Image(label="Generated Image"),
|
| 122 |
title="Generate Image",
|