"src/vscode:/vscode.git/clone" did not exist on "97a51b0c7d483cdf13ea878a987f9aa1c9eecc91"
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