search

showlab / DatasetDM

[NeurIPS2023] DatasetDM:Synthesizing Data with Perception Annotations Using Diffusion Models
https://weijiawu.github.io/DatasetDM_page/
299 stars 12 forks source link

In the train code，file not found， #5

Closed aiforworlds closed 1 year ago

aiforworlds commented 1 year ago

Thanks for your great work. When I run the training code ./tools/train_depth_NYU.py, I find that the file that needs to be imported cannot be found， in ./tools/train_depth_NYU.py file line 25 from model.diffusers.models.unet_2d_condition import UNet2DConditionModel and in model/unet.py line9 from model.diffusers.models.unet_2d_condition import UNet2DConditionModel

Looking forward to your reply.

weijiawu commented 1 year ago

Thanks for your attention.

One possible solution like: cd model git clone https://github.com/huggingface/diffusers.git

aiforworlds commented 1 year ago

Thanks for your reply. I did it according to your suggestion, but still have the problem。 diffuser code cann't find models folder, from https://github.com/huggingface/diffusers.git sofrom model.diffusers.models.unet_2d_condition import UNet2DConditionModelstill error

weijiawu commented 1 year ago

Do not use the whole file, just copy the diffusers file https://github.com/huggingface/diffusers/tree/main/src/diffusers to the ./model path.

aiforworlds commented 1 year ago

Thank you very much. The problem has been solved.