Spaces:
Running
on
A10G
Running
on
A10G
Edit attention map selection upon num_edit_tokens
#15
by
binhle0209
- opened
In line 932-935, selecting attention map range should be upon the num_edit_tokens[batch_size*c], as the num_edit_tokens having each concept repeated batch_size times.