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

Unconditional Diffusion Example without Grayscale #14

Closed codeHorasan closed 5 months ago

codeHorasan commented 6 months ago

How to run the Unconditional Diffusion example on custom dataset without grayscale? It returns error whenever the dataset is not grayscaled

cabralpinto commented 6 months ago

@codeHorasan Could you provide the code snippet in question and the error you get? The library should work for any data type but you might need to reshape it depending on the network and the noise type you choose.

cabralpinto commented 5 months ago

Will close due to inactivity.