ldm add plms sampler
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user