Add `clip_sample=False` to scheduler to make model compatible with DDIM.

End of training