From a49cb59dd06ff9b2b3b2dd8b9a53e4a0059206dc Mon Sep 17 00:00:00 2001 From: xiaomile <15622388695@163.com> Date: Sat, 7 Oct 2023 14:10:59 +0800 Subject: [PATCH] Adding support for Consistency Models rerun circle check --- .../models/editors/consistency_models/consistencymodel_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/mmagic/models/editors/consistency_models/consistencymodel_utils.py b/mmagic/models/editors/consistency_models/consistencymodel_utils.py index 4774f374d..73354f8e6 100644 --- a/mmagic/models/editors/consistency_models/consistencymodel_utils.py +++ b/mmagic/models/editors/consistency_models/consistencymodel_utils.py @@ -78,7 +78,6 @@ def karras_sample( steps, sigma_min, sigma_max, rho, device=device) x_T = generator.randn(*shape, device=device) * sigma_max - sample_fn = get_sample_fn(sampler) if sampler in ['heun', 'dpm']: