"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "02a0b43014ac333a169e99d76aaba023a316e384"
Specify dtype=torch.bool to avoid xla error (#31191)
The StoppingCriteriaList allocates is_done without specifying dtype=torch.bool. On XLA this allocates a float tensor and causes a failure on the following line: is_done = is_done | criteria(input_ids, scores, **kwargs) by attempting to OR float with bool.
Showing
Please register or sign in to comment