search

crowsonkb / k-diffusion

Karras et al. (2022) diffusion models for PyTorch
MIT License
2.26k stars 372 forks source link

Simple evaluation script #41

Open pcuenca opened 1 year ago

pcuenca commented 1 year ago

We wrote this for fastai but might be useful here too :)

Note: it requires fastcore.

We could extend it, if desired, to download models from Hugging Face Hub or W&B artifacts.