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

Separate python dependencies into optional groups #38

Open luke-carlson opened 1 week ago

luke-carlson commented 1 week ago

Some project dependencies are only needed for the web page, some are only needed during training, etc. It could be worth breaking the dependency list into groups so that users don't need to install all of them just to get started with using the repo

[project.optional-dependencies]
data_prep = [
    "img2dataset"
]
web_demo = [
    ...
]
training = [
   ...
]
pedroborgescruz commented 4 days ago

@aaliyahnl and I are working on this!