search

sail-sg / MDT

Masked Diffusion Transformer is the SOTA for image synthesis. (ICCV 2023)
Apache License 2.0
500 stars 35 forks source link

Module Import error in train_util.py #11

Closed Bhavay-2001 closed 1 year ago

Bhavay-2001 commented 1 year ago

Hi, I'm trying to run the file train_util.py you provided but it throws an error - "ModuleNotFoundError: No module named 'diffusers'". I read the code and you have basically imported AutoEncoder.py from the diffusers module of the hugging face repository but how it is done? How have you made the import statement work since you haven't cloned the hugging face GitHub repository.

gasvn commented 1 year ago

Just install diffusers with pip install diffusers