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

Potential update to pyproject.toml to get web demos to run? #34

Open emmagarr opened 1 week ago

emmagarr commented 1 week ago

I'd been getting a segfault trying to run the web demo, but I was finally able to run it successfully after upgrading torch and torchvision. I currently have torch 2.5.1 and torchvision 0.20.1. Would you want to reflect this pyproject.toml? If not, maybe we can put a note in the ReadMe for people who are getting the same error.

levinkhho commented 1 week ago

https://github.com/apple/ml-mdm/pull/35