feature: change default sampler to k_dpmpp_2m, which is more than twice as fast

pull/88/head
Bryce 2 years ago committed by Bryce Drennan
parent 0f02fc587c
commit 584e0d9df0

@ -224,11 +224,14 @@ docker run -it --gpus all -v $HOME/.cache/huggingface:/root/.cache/huggingface -
## ChangeLog ## ChangeLog
**6.0.0**
- feature: new default sampler makes image generation mor than twice as fast
- feature: added `DPM++ 2S a` and `DPM++ 2M` samplers. - feature: added `DPM++ 2S a` and `DPM++ 2M` samplers.
- feature: improve progress image logging - feature: improve progress image logging
- fix: fix bug with `--show-work` - fix: fix bug with `--show-work`. fixes #84
- fix: add workaround for pytorch bug affecting MacOS users using the new `DPM++ 2S a` and `DPM++ 2M` samplers. - fix: add workaround for pytorch bug affecting MacOS users using the new `DPM++ 2S a` and `DPM++ 2M` samplers.
- fix: add workaround for pytorch mps bug affecting `k_dpm_fast` sampler. fixes #75 - fix: add workaround for pytorch mps bug affecting `k_dpm_fast` sampler. fixes #75
- fix: larger image sizes now work on MacOS. fixes #8
**5.0.0** **5.0.0**
- feature: 🎉 inpainting support using new inpainting model from RunwayML. It works really well! (Unfortunately it requires a HuggingFace token). - feature: 🎉 inpainting support using new inpainting model from RunwayML. It works really well! (Unfortunately it requires a HuggingFace token).

@ -52,7 +52,7 @@ logger = logging.getLogger(__name__)
) )
@click.option( @click.option(
"--steps", "--steps",
default=40, default=15,
type=int, type=int,
show_default=True, show_default=True,
help="How many diffusion steps to run. More steps, more detail, but with diminishing returns", help="How many diffusion steps to run. More steps, more detail, but with diminishing returns",
@ -73,7 +73,7 @@ logger = logging.getLogger(__name__)
@click.option( @click.option(
"--sampler-type", "--sampler-type",
"--sampler", "--sampler",
default="plms", default="k_dpmpp_2m",
type=click.Choice(SAMPLER_TYPE_OPTIONS), type=click.Choice(SAMPLER_TYPE_OPTIONS),
help="What sampling strategy to use", help="What sampling strategy to use",
) )

Loading…
Cancel
Save