Skip to content

GitLab

Unverified Commit f064b3bf authored by Aryan's avatar Aryan Committed by GitHub
Browse files

Remove print statement in SCM Scheduler (#11836) 

remove print
parent 3b079ec3
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/diffusers/schedulers/scheduling_scm.py
View file @ f064b3bf
...@@ -168,7 +168,6 @@ class SCMScheduler(SchedulerMixin, ConfigMixin): ...@@ -168,7 +168,6 @@ class SCMScheduler(SchedulerMixin, ConfigMixin):
else: else:
# max_timesteps=arctan(80/0.5)=1.56454 is the default from sCM paper, we choose a different value here # max_timesteps=arctan(80/0.5)=1.56454 is the default from sCM paper, we choose a different value here
self.timesteps = torch.linspace(max_timesteps, 0, num_inference_steps + 1, device=device).float() self.timesteps = torch.linspace(max_timesteps, 0, num_inference_steps + 1, device=device).float()
print(f"Set timesteps: {self.timesteps}")
self._step_index = None self._step_index = None
self._begin_index = None self._begin_index = None
......
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