| from dataclasses import dataclass | |
| from PIL import Image | |
| from typing import List | |
| import torch | |
| class VyroParams: | |
| latents: torch.Tensor | |
| user_prompt: str | |
| user_neg_prompt: str | |
| mode: str | |
| cfg: float | |
| batch_size: int | |
| steps: int | |
| width: int | |
| height: int | |
| seed: int | |
| denoise: float | |
| stage1_strength: float | |
| stage2_strength: float | |
| efficiency_multiplier: float | |
| style: str = '' | |
| final_positive_prompt: str = '' | |
| final_negative_prompt: str = '' | |
| is_raw: bool = False | |
| MULTIPLIER = [1.0,1.33, 2.0] | |
| STATES = ["disabled", "enabled"] | |
| ALLOWED = ["allow", "deny"] | |
| MODE = ["t2i", "i2i"] | |
| STYLES = [ | |
| "3d render", | |
| "abstract art", | |
| "anime", | |
| "architecture", | |
| "cinematic", | |
| "conceptual art", | |
| "dark fantasy", | |
| "fantasy realism", | |
| "fashion", | |
| "graffiti", | |
| "illustration", | |
| "interior design", | |
| "logo", | |
| "painting", | |
| "photography", | |
| "portrait photography", | |
| "poster", | |
| "product", | |
| "sticker", | |
| "surrealism", | |
| "typography", | |
| "ukiyo-e", | |
| "vector design", | |
| "vibrant digital artwork", | |
| "watercolor", | |
| "wildlife photography" | |
| ] | |
| PARAMS = [ | |
| "latents", | |
| "user_prompt", | |
| "user_neg_prompt", | |
| "mode", | |
| "cfg", | |
| "steps", | |
| "width", | |
| "height", | |
| "seed", | |
| "denoise", | |
| "stage1_strength", | |
| "stage2_strength", | |
| "efficiency_multiplier", | |
| "style", | |
| "final_positive_prompt", | |
| "final_negative_prompt", | |
| "is_raw" | |
| ] | |