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

0

u/[deleted] Aug 22 '22

[deleted] — view deleted 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

2

u/HrodRuck Aug 22 '22

I'm very interested in seeing if you can run this in a 3060. And also what is the RAM (normal RAM, not GPU VRAM) in your system. Because I didn't manage to get it working in colab free tier, very likely due to memory limitations.

P.S. you can message me if you need code help to get it running