-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Closed
Labels
bugSomething isn't workingSomething isn't workingstaleIssues that haven't received updatesIssues that haven't received updates
Description
Describe the bug
In this line, single_image_embeds = torch.stack([single_image_embeds] * num_images_per_prompt, dim=0) will add a new dimension to single_image_embeds,making the image_embedding has 4 dimensions. Would be better to use torch.cat()?
Reproduction
It won't cause errors for now since the embedding is reshaped in attention processor. But current format looks strange.
Logs
No response
System Info
Latest
Who can help?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
bugSomething isn't workingSomething isn't workingstaleIssues that haven't received updatesIssues that haven't received updates