Adding support for Consistency Models

rerun circle check
open-mmlab · Oct 7, 2023 · a49cb59 · a49cb59
1 parent 864b3aa
commit a49cb59
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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']: