ldm add plms sampler

This commit is contained in:
Qing
2022-06-12 13:14:17 +08:00
parent 55197f2209
commit 35b92ba9de
11 changed files with 478 additions and 207 deletions

View File

@@ -9,8 +9,14 @@ class HDStrategy(str, Enum):
CROP = 'Crop'
class LDMSampler(str, Enum):
ddim = 'ddim'
plms = 'plms'
class Config(BaseModel):
ldm_steps: int
ldm_sampler: str
hd_strategy: str
hd_strategy_crop_margin: int
hd_strategy_crop_trigger_size: int