diffusers
0db766ba - [DDPMScheduler] Load `alpha_cumprod` to device to avoid redundant data movement. (#6704)

Commit
1 year ago
[DDPMScheduler] Load `alpha_cumprod` to device to avoid redundant data movement. (#6704) * load cumprod tensor to device Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com> * fixing ci Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com> * make fix-copies Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com> --------- Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com>
Author
Parents
Loading