open-genie
open-genie copied to clipboard
A small shape bug
https://github.com/myscience/open-genie/blob/732b9f9b746f18fff1a0fb22f83638224f2f7cc6/genie/module/attention.py#L290C26-L290C37
b, *t, h, w, c = video.shape
should be:
b, *t, h, w, c = inp.shape