search

jina-ai / dalle-flow

🌊 A Human-in-the-Loop workflow for creating HD images from text
grpcs://dalle-flow.dev.jina.ai
2.83k stars 209 forks source link

Add new k-diffusion samplers to stable executor #132

Closed AmericanPresidentJimmyCarter closed 1 year ago

AmericanPresidentJimmyCarter commented 1 year ago

This adds the newest samplers from k-diffusion into the stable executor along with the ability to choose a default noise schedule for the diffusion unet steps.

From k-diffusion: "DPM-Solver++(2S) and (2M) are implemented now too for improved quality with low numbers of steps."

AmericanPresidentJimmyCarter commented 1 year ago

This also fixes a Dockerfile issue due to jax being too low of a version for transformers. The jax version has been bumped to 0.3.24.

samsja commented 1 year ago

Can you do the jax bump in a separate PR ?

AmericanPresidentJimmyCarter commented 1 year ago

Can you do the jax bump in a separate PR ?

OK, I will split them up later today.

AmericanPresidentJimmyCarter commented 1 year ago

@samsja It has been split out.

AmericanPresidentJimmyCarter commented 1 year ago

Also to test you can sample with 'dpm_fast', 'dpm_adaptive', 'dpmpp_2s_ancestral', and 'dpmpp_2m'. Valid values for noiser are None, 'karras', 'exponential', and 'vp'.