logits_processor=logits_processor error
#1
by
RobertLee2Future
- opened
In the demo case, the setting of logits_processor may lead to the following error
'''
---> 50 height = self.height[batch_id] if self.height.shape[0] > 1 else self.height[0]
51 width = self.width[batch_id] if self.width.shape[0] > 1 else self.width[0]
53 offset = input_ids.shape[0] - self.offset_cache[batch_id]
IndexError: tuple index out of range
'''
Removing the setting is fixed