generated training dict missing some keys.

frank-xwang / InstanceDiffusion

[CVPR 2024] Code release for "InstanceDiffusion: Instance-level Control for Image Generation"

https://people.eecs.berkeley.edu/~xdwang/projects/InstDiff/

Apache License 2.0

515 stars 29 forks source link

generated training dict missing some keys. #5

Closed zxduan90 closed 9 months ago

zxduan90 commented 9 months ago

The jsondataset decode func needs "image_embedding_before", "text_embedding_before", "image_embedding_after", "text_embedding_after", but only "text_embedding_before" is generated.

frank-xwang commented 9 months ago

Hi thank you for noticing it! I've updated the codebase and addressed the bug. Please don't hesitate to reopen the issue if you encounter any further problems. :-)