search

pixeli99 / SVD_Xtend

Stable Video Diffusion Training Code and Extensions.
553 stars 51 forks source link

Fix timesteps device #5

Closed BlakeOne closed 8 months ago

BlakeOne commented 8 months ago

Set the device of the timesteps to that of the accelerator. This way the timesteps and the model weights are on the same device. This fix enabled me to finetune the unet without errors. :)

pixeli99 commented 8 months ago

Thanks :), they have been merged.