https://github.com/congvvc/HyperSeg/blob/5cfcf5259accb0c7ff1b6ca3f7ea7457bb7518c5/hyperseg/model/language_model/llava_phi.py#L1333 I think I found a bug in your codes. Since the 'return' statement exists inside the for loop, the code does not work for a batch size greater than 1.
HyperSeg/hyperseg/model/language_model/llava_phi.py
Line 1333 in 5cfcf52
I think I found a bug in your codes.
Since the 'return' statement exists inside the for loop, the code does not work for a batch size greater than 1.