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>

References

#6704 - [DDPMScheduler] Load `alpha_cumprod` to device to avoid redundant data movement.

Author

woshiyyya

Parents

8e946635

Files8

src/diffusers/schedulers
- scheduling_ddim.py
- scheduling_ddim_parallel.py
- scheduling_ddpm.py
- scheduling_ddpm_parallel.py
- scheduling_lcm.py
- scheduling_pndm.py
- scheduling_sasolver.py
- scheduling_unclip.py

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

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