TypeError: UNet2DConditionModel.forward() got an unexpected keyword argument 'objs'

OPPO-Mente-Lab / attention-mask-control

code for paper "Compositional Text-to-Image Synthesis with Attention Map Control of Diffusion Models"

MIT License

39 stars 5 forks source link

TypeError: UNet2DConditionModel.forward() got an unexpected keyword argument 'objs' #2

Closed fhlt closed 1 year ago

Wrch1994 commented 1 year ago

Hello, we have fixed some bugs in the training code in the latest commit. After testing, this version of the code can be used for training normally. Please pull the latest code in the master branch and set up the environment according to the updated environment.yaml to prevent any issues with the versions of Python libraries.