Update video node to support more parameters and make it more dynamic
This commit is contained in:
@@ -0,0 +1,36 @@
|
||||
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}).",
|
||||
)
|
||||
Reference in New Issue
Block a user