Spaces:
Running
Running
File size: 556 Bytes
6bcacf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from typing import Optional
from pydantic import BaseModel
from constants import LCM_DEFAULT_MODEL
class LCMDiffusionSetting(BaseModel):
lcm_model_id: str = LCM_DEFAULT_MODEL
prompt: str = ""
image_height: Optional[int] = 512
image_width: Optional[int] = 512
inference_steps: Optional[int] = 4
guidance_scale: Optional[float] = 8
number_of_images: Optional[int] = 1
seed: Optional[int] = -1
use_openvino: bool = False
use_seed: bool = False
use_offline_model: bool = False
use_safety_checker: bool = True
|