search

lucidrains / muse-maskgit-pytorch

Implementation of Muse: Text-to-Image Generation via Masked Generative Transformers, in Pytorch
MIT License
850 stars 81 forks source link

ValueError: You can only pass one `DistributedDataParallelKwargs` in `kwargs_handler`. #31

Open Zxlan opened 1 year ago

Zxlan commented 1 year ago

I installed the code and its dependencies. When I train VAE , it fails :

图片

I'm using Python 3.9. Is that an ok version to use? I installed the muse-maskgit-pytorch by pip install muse-maskgit-pytorch.

lucidrains commented 1 year ago

@Zxlan can i see your full script?

lucidrains commented 1 year ago

@Zxlan want to try 0.1.4 and see if that resolves your issue? (and if not, share your script)