From 678ac2ffc9056f79991c287ea845d9add16ff186 Mon Sep 17 00:00:00 2001 From: huchenlei Date: Fri, 20 Dec 2024 09:48:57 -0800 Subject: [PATCH] nit --- comfy/extra_samplers/uni_pc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/extra_samplers/uni_pc.py b/comfy/extra_samplers/uni_pc.py index 6062c7795..b61baaa8e 100644 --- a/comfy/extra_samplers/uni_pc.py +++ b/comfy/extra_samplers/uni_pc.py @@ -2,6 +2,7 @@ import torch import math +import logging from tqdm.auto import trange @@ -474,7 +475,7 @@ class UniPC: return self.multistep_uni_pc_vary_update(x, model_prev_list, t_prev_list, t, order, **kwargs) def multistep_uni_pc_vary_update(self, x, model_prev_list, t_prev_list, t, order, use_corrector=True): - print(f'using unified predictor-corrector with order {order} (solver type: vary coeff)') # noqa: T201 + logging.info(f'using unified predictor-corrector with order {order} (solver type: vary coeff)') ns = self.noise_schedule assert order <= len(model_prev_list)