File size: 1,714 Bytes
6fecfbe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
from dataclasses import dataclass
from PIL import Image
from typing import List
import torch
@dataclass
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"
]
|