diffusion feature may be wrongly used

[centosrhel]

DragAnything/demo.py

Line 396 in 438d2be

ids_embedding[coord[0], coord[1]] = id_feature[0]

Here you are assigning a single floating-point number to a vector, resulting in a vector with all the same number. Is it a typo error?

[centosrhel]

@weijiawu Could you please check this line of code? And was the open model also trained as so?

[zhw-zhang]

yes，I also noticed this problem...