big update
This commit is contained in:
17
lama_cleaner/schema.py
Normal file
17
lama_cleaner/schema.py
Normal file
@@ -0,0 +1,17 @@
|
||||
from enum import Enum
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
|
||||
class HDStrategy(str, Enum):
|
||||
ORIGINAL = 'Original'
|
||||
RESIZE = 'Resize'
|
||||
CROP = 'Crop'
|
||||
|
||||
|
||||
class Config(BaseModel):
|
||||
ldm_steps: int
|
||||
hd_strategy: str
|
||||
hd_strategy_crop_margin: int
|
||||
hd_strategy_crop_trigger_size: int
|
||||
hd_strategy_resize_limit: int
|
||||
Reference in New Issue
Block a user