CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

batch size

Browse Source
This commit is contained in:
Haoming 2026-05-08 11:54:41 +08:00
parent a8668a0de0
commit 3341f41adc
2 changed files with 32 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules_forge/main_entry.py
View File

@ -18,7 +18,7 @@ from modules import (
shared_items,
ui_common,
)
from modules_forge.presets import PresetArch, use_distill, use_shift
from modules_forge.presets import PresetArch, is_video, use_distill, use_shift
logger = logging.getLogger("ui_models")
setup_logger(logger)
@ -285,7 +285,10 @@ def on_preset_change(preset: str):
else:
d_args = {"visible": False}
batch_args = {"minimum": 1, "maximum": 241, "step": 16, "label": "Frames", "value": 1} if preset == "wan" else {"minimum": 1, "maximum": 8, "step": 1, "label": "Batch Size", "value": 1}
if (fps := is_video(preset)) > 1:
batch_args = {"minimum": 1, "maximum": fps * 15 + 1, "step": fps, "label": "Frames", "value": getattr(shared.opts, f"{preset}_batch_size", 1)}
else:
batch_args = {"minimum": 1, "maximum": 8, "step": 1, "label": "Batch Size", "value": getattr(shared.opts, f"{preset}_batch_size", 1)}
return [
# ui_checkpoint, ui_vae, ui_forge_unet_dtype

27
modules_forge/presets.py
View File

@ -83,6 +83,10 @@ SHIFT = {
PresetArch.ernie: 3.0,
}
FRAMES = {
PresetArch.wan.name: 16,
}
def use_distill(arch: str) -> bool:
return arch in [preset.name for preset in DISTILL.keys()]
@ -92,6 +96,10 @@ def use_shift(arch: str) -> bool:
return arch in [preset.name for preset in SHIFT.keys()]
def is_video(arch: str) -> int:
return FRAMES.get(arch, 1)
def register(options_templates: dict):
from gradio import Dropdown, Slider
@ -189,6 +197,25 @@ def register(options_templates: dict):
)
)
if (fps := FRAMES.get(arch.name, 1)) > 1:
options_templates.update(
options_section(
(f"ui_{name}", name.upper(), "presets"),
{
f"{name}_batch_size": OptionInfo(1, "Frames", Slider, {"minimum": 1, "maximum": fps * 15 + 1, "step": fps}),
},
)
)
else:
options_templates.update(
options_section(
(f"ui_{name}", name.upper(), "presets"),
{
f"{name}_batch_size": OptionInfo(1, "Batch Size", Slider, {"minimum": 1, "maximum": 8, "step": 1}),
},
)
)
options_templates.update(
options_section(
(f"ui_{name}", name.upper(), "presets"),