add DPM-Solver++(2M) SDE from new k-diffusion (31545abe) · Commits · github_fork / Stable Diffusion Webui

launch.py

+1 −1

Original line number	Diff line number	Diff line
		@@ -236,7 +236,7 @@ def prepare_environment():

		stable_diffusion_commit_hash = os.environ.get('STABLE_DIFFUSION_COMMIT_HASH', "cf1d67a6fd5ea1aa600c4df58e5b47da45f6bdbf")
		taming_transformers_commit_hash = os.environ.get('TAMING_TRANSFORMERS_COMMIT_HASH', "24268930bf1dce879235a7fddd0b2355b84d7ea6")
		k_diffusion_commit_hash = os.environ.get('K_DIFFUSION_COMMIT_HASH', "5b3af030dd83e0297272d861c19477735d0317ec")
		k_diffusion_commit_hash = os.environ.get('K_DIFFUSION_COMMIT_HASH', "c9fe758757e022f05ca5a53fa8fac28889e4f1cf")
		codeformer_commit_hash = os.environ.get('CODEFORMER_COMMIT_HASH', "c5b4593074ba6214284d6acd5f1719b6c5d739af")
		blip_commit_hash = os.environ.get('BLIP_COMMIT_HASH', "48211a1594f1321b00f14c9f7a5b4813144b2fb9")

modules/sd_samplers_kdiffusion.py

+8 −6

Original line number	Diff line number	Diff line
		@@ -19,7 +19,8 @@ samplers_k_diffusion = [
		('DPM2 a', 'sample_dpm_2_ancestral', ['k_dpm_2_a'], {'discard_next_to_last_sigma': True, "uses_ensd": True}),
		('DPM++ 2S a', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a'], {"uses_ensd": True, "second_order": True}),
		('DPM++ 2M', 'sample_dpmpp_2m', ['k_dpmpp_2m'], {}),
		('DPM++ SDE', 'sample_dpmpp_sde', ['k_dpmpp_sde'], {"second_order": True}),
		('DPM++ SDE', 'sample_dpmpp_sde', ['k_dpmpp_sde'], {"second_order": True, "brownian_noise": True}),
		('DPM++ 2M SDE', 'sample_dpmpp_2m_sde', ['k_dpmpp_2m_sde_ka'], {"brownian_noise": True}),
		('DPM fast', 'sample_dpm_fast', ['k_dpm_fast'], {"uses_ensd": True}),
		('DPM adaptive', 'sample_dpm_adaptive', ['k_dpm_ad'], {"uses_ensd": True}),
		('LMS Karras', 'sample_lms', ['k_lms_ka'], {'scheduler': 'karras'}),
		@@ -27,7 +28,8 @@ samplers_k_diffusion = [
		('DPM2 a Karras', 'sample_dpm_2_ancestral', ['k_dpm_2_a_ka'], {'scheduler': 'karras', 'discard_next_to_last_sigma': True, "uses_ensd": True, "second_order": True}),
		('DPM++ 2S a Karras', 'sample_dpmpp_2s_ancestral', ['k_dpmpp_2s_a_ka'], {'scheduler': 'karras', "uses_ensd": True, "second_order": True}),
		('DPM++ 2M Karras', 'sample_dpmpp_2m', ['k_dpmpp_2m_ka'], {'scheduler': 'karras'}),
		('DPM++ SDE Karras', 'sample_dpmpp_sde', ['k_dpmpp_sde_ka'], {'scheduler': 'karras', "second_order": True}),
		('DPM++ SDE Karras', 'sample_dpmpp_sde', ['k_dpmpp_sde_ka'], {'scheduler': 'karras', "second_order": True, "brownian_noise": True}),
		('DPM++ 2M SDE Karras', 'sample_dpmpp_2m_sde', ['k_dpmpp_2m_sde_ka'], {'scheduler': 'karras', "brownian_noise": True}),
		]

		samplers_data_k_diffusion = [
		@@ -228,7 +230,7 @@ class KDiffusionSampler:
		self.sampler_noises = None
		self.stop_at = None
		self.eta = None
		self.config = None
		self.config = None # set by the function calling the constructor
		self.last_latent = None
		self.s_min_uncond = None

		@@ -337,7 +339,7 @@ class KDiffusionSampler:
		if 'sigmas' in parameters:
		extra_params_kwargs['sigmas'] = sigma_sched

		if self.funcname == 'sample_dpmpp_sde':
		if self.config.options.get('brownian_noise', False):
		noise_sampler = self.create_noise_sampler(x, sigmas, p)
		extra_params_kwargs['noise_sampler'] = noise_sampler

		@@ -373,7 +375,7 @@ class KDiffusionSampler:
		else:
		extra_params_kwargs['sigmas'] = sigmas

		if self.funcname == 'sample_dpmpp_sde':
		if self.config.options.get('brownian_noise', False):
		noise_sampler = self.create_noise_sampler(x, sigmas, p)
		extra_params_kwargs['noise_sampler'] = noise_sampler

Admin message