apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner
https://machinelearning.apple.com/research/matryoshka-diffusion-models
MIT License
448 stars 31 forks source link

Create separate dependency lists for cpu vs gpu, eg "pip install .[cpu]" vs "pip install .[gpu]" #40

Open luke-carlson opened 1 week ago