sd make change sampler work

This commit is contained in:
Qing
2022-09-22 12:38:32 +08:00
parent 047474ab84
commit e1fb0030d1
3 changed files with 46 additions and 19 deletions

View File

@@ -14,6 +14,11 @@ class LDMSampler(str, Enum):
plms = 'plms'
class SDSampler(str, Enum):
ddim = 'ddim'
pndm = 'pndm'
class Config(BaseModel):
ldm_steps: int
ldm_sampler: str = LDMSampler.plms
@@ -35,6 +40,6 @@ class Config(BaseModel):
sd_strength: float = 0.75
sd_steps: int = 50
sd_guidance_scale: float = 7.5
sd_sampler: str = 'ddim' # ddim/pndm
sd_sampler: str = SDSampler.ddim
# -1 mean random seed
sd_seed: int = 42