Nodes/Parametric Studio/PSN Batch Size Optimizer
ComfyUI Node

PSN Batch Size Optimizer

Calculate the optimal image batch size from available VRAM and model size.

By parametricparadigm·Created 4 days ago·Updated about 3 hours ago· 1
PSN Batch Size Optimizer
    • optimal_batch
    • estimated_vram_usage
    • reason
    ◄vram_available_gb24.0►
    ◄model_size_gb6.0►
    ◄per_image_vram_gb1.00►
    ◄min_batch1►
    ◄max_batch8►
    ◄safety_margin_pct0.10►
    CategoryParametric/Flow

    Inputs (6)

    NameTypeDefaultDescription
    vram_available_gbFLOAT24.00–1000000Vram available gb. From 0 to 1e+06. Default 24.0.
    model_size_gbFLOAT6.00–1000000Model size gb. From 0 to 1e+06. Default 6.0.
    per_image_vram_gbFLOAT1.000.001–1000000Per image vram gb. From 0.001 to 1e+06. Default 1.0.
    min_batchINT11–1024Min batch. From 1 to 1024. Default 1.
    max_batchINT81–1024Max batch. From 1 to 1024. Default 8.
    safety_margin_pctFLOAT0.100–0.999Safety margin pct. From 0 to 0.999. Default 0.1.

    Outputs (3)

    NameTypeDescription
    optimal_batchINT—
    estimated_vram_usageFLOAT—
    reasonSTRING—