"git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "3c760180440ed1126f42c59914f5df0185e33983"
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
......@@ -168,7 +168,6 @@ class SCMScheduler(SchedulerMixin, ConfigMixin):
else:
# 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()
print(f"Set timesteps: {self.timesteps}")
self._step_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