Spaces:
Running
Running
File size: 391 Bytes
c7f95a6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from pydantic import BaseModel
from enum import Enum
from paths import FastStableDiffusionPaths
class ImageFormat(str, Enum):
"""Image format"""
JPEG = "jpeg"
PNG = "png"
class GeneratedImages(BaseModel):
path: str = FastStableDiffusionPaths.get_results_path()
format: str = ImageFormat.PNG.value.upper()
save_image: bool = True
save_image_quality: int = 90
|