alembics / disco-diffusion

Other
7.48k stars 1.13k forks source link

Add option to run with CPU #71

Closed OodavidsinoO closed 2 years ago

OodavidsinoO commented 2 years ago

Skip running model.convert_to_fp16() if use CPU. Run without error. image

aletts commented 2 years ago

Hi. Thanks! The change is really cool, and now I see how to do it and it make sense.

However.. we need to maintain the commented data present in disco.py. That metadata is what allows is to generate the .ipynb file from disco.py using @MSFTserver 's conversion tool. So it can't be merged until there's a PR which does that.

OodavidsinoO commented 2 years ago

Working on it. Thanks for the advice.