File size: 497 Bytes
3dabe4a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from scripts.iib.tool import omit
class ImageGenerationParams:
def __init__(self, meta: dict = {}, pos_prompt: list = [], extra: dict = {}) -> None:
self.meta = meta
self.pos_prompt = pos_prompt
self.extra = omit(extra, ["meta", "pos_prompt"])
class ImageGenerationInfo:
def __init__(
self,
raw_info: str = "",
params: ImageGenerationParams = ImageGenerationParams(),
):
self.raw_info = raw_info
self.params = params
|