Nymbo commited on
Commit
f437d5e
·
verified ·
1 Parent(s): 516f417

Update Modules/Generate_Image.py

Browse files
Files changed (1) hide show
  1. Modules/Generate_Image.py +4 -4
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",