stable-dreambooth icon indicating copy to clipboard operation
stable-dreambooth copied to clipboard

error whie running train.py

Open mohitm1994 opened this issue 2 years ago • 1 comments

File "train.py", line 206, in train_loop(config, model, noise_scheduler, optimizer, train_dataloader) File "train.py", line 131, in train_loop noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps.cpu().numpy()) File "/HPS/EgofaceTrial/work/anaconda3/envs/stable-diffusion/lib/python3.8/site-packages/diffusers/schedulers/scheduling_ddpm.py", line 303, in add_noise timesteps = timesteps.to(original_samples.device) AttributeError: 'numpy.ndarray' object has no attribute 'to'

mohitm1994 avatar Nov 07 '22 21:11 mohitm1994

Need to change to: noisy_latents = noise_scheduler.add_noise(latents, noise, timesteps)

Arthur-Lee avatar Oct 03 '23 13:10 Arthur-Lee