Commit ce0aeb10 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove print.

parent 5489d5af
...@@ -710,7 +710,6 @@ class UniPC: ...@@ -710,7 +710,6 @@ class UniPC:
if method == 'multistep': if method == 'multistep':
assert steps >= order assert steps >= order
# timesteps = self.get_time_steps(skip_type=skip_type, t_T=t_T, t_0=t_0, N=steps, device=device) # 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 assert timesteps.shape[0] - 1 == steps
# with torch.no_grad(): # with torch.no_grad():
for step_index in trange(steps + 1): for step_index in trange(steps + 1):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment