Remove print.

This commit is contained in:
comfyanonymous 2023-02-11 03:41:40 -05:00
parent 5489d5af04
commit ce0aeb109e

View File

@ -710,7 +710,6 @@ class UniPC:
if method == 'multistep':
assert steps >= order
# timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device)
print(timesteps)
assert timesteps.shape[0] - 1 == steps
# with torch.no_grad():
for step_index in trange(steps + 1):