37 lines
990 B
Python
37 lines
990 B
Python
from comfy_api.latest import io
|
|
|
|
from src.settings.base import Resolutions, Settings, Sizes
|
|
|
|
IMAGE_SIZES = [
|
|
Sizes._21_9,
|
|
Sizes._16_9,
|
|
Sizes._3_2,
|
|
Sizes._4_3,
|
|
Sizes._5_4,
|
|
Sizes._1_1,
|
|
Sizes._4_5,
|
|
Sizes._3_4,
|
|
Sizes._2_3,
|
|
Sizes._9_16,
|
|
]
|
|
IMAGE_RESOLUTIONS = [Resolutions._480, Resolutions._720, Resolutions._1080, Resolutions._4K]
|
|
|
|
DEFAULT_MIN_IMAGES = 1
|
|
|
|
|
|
class ImageSettings(Settings):
|
|
"""A model's image generation bounds: which of the master SIZE/RESOLUTIONS
|
|
options it supports, and how many images it can generate per call."""
|
|
|
|
MIN_IMAGES: int = DEFAULT_MIN_IMAGES
|
|
MAX_IMAGES: int
|
|
|
|
def num_images_input(self, default=None):
|
|
return io.Int.Input(
|
|
"num_images",
|
|
default=default if default is not None else self.MIN_IMAGES,
|
|
min=self.MIN_IMAGES,
|
|
max=self.MAX_IMAGES,
|
|
tooltip=f"Number of images to generate at once ({self.MIN_IMAGES}-{self.MAX_IMAGES}).",
|
|
)
|