IOPaint/lama_cleaner/schema.py

24 lines
412 B
Python
Raw Normal View History

2022-04-15 18:11:51 +02:00
from enum import Enum
from pydantic import BaseModel
class HDStrategy(str, Enum):
ORIGINAL = 'Original'
RESIZE = 'Resize'
CROP = 'Crop'
2022-06-12 07:14:17 +02:00
class LDMSampler(str, Enum):
ddim = 'ddim'
plms = 'plms'
2022-04-15 18:11:51 +02:00
class Config(BaseModel):
ldm_steps: int
2022-06-12 07:14:17 +02:00
ldm_sampler: str
2022-04-15 18:11:51 +02:00
hd_strategy: str
hd_strategy_crop_margin: int
hd_strategy_crop_trigger_size: int
hd_strategy_resize_limit: int