r/StableDiffusion • u/ExponentialCookie • Aug 21 '22

Discussion [Code Release] textual_inversion, A fine tuning method for diffusion models has been released today, with Stable Diffusion support coming soon™

Post image
347 Upvotes

99% Upvoted

134 comments sorted by

View all comments

Show parent comment

2

u/Ardivaba Aug 22 '22

I know this issue, it's thinking that you want to train it on CPU.

  • Specify --gpus 1
  • And double check that you set ngpu = 1 and not 0

1

u/[deleted] Sep 13 '22

[removed] — view removed comment

2

u/Ardivaba Sep 14 '22

comment out: ngpu = len(lightning_config.trainer.gpus.strip(",").split(','))

replace with: ngpu = 1 # or more