search

cabralpinto / modular-diffusion

Python library for designing and training your own Diffusion Models with PyTorch.
https://cabralpinto.github.io/modular-diffusion/
MIT License
266 stars 12 forks source link

No diffusion.Model file #16

Closed gagewrye closed 3 months ago

gagewrye commented 3 months ago

The Model file is missing, making the repo unusable.

cabralpinto commented 3 months ago

Hi @gagewrye, thanks for the issue. This is by design. Model is a class declared inside __init__.py. Does this solve your problem?

gagewrye commented 3 months ago

Thanks for the quick reply! It wasn't initializing at first for some reason and I couldn't use the Model class. It's working now though.