transformers
24d59c79 - Use `torch.bool` instead of `torch.int64` for non-persistant causal mask buffer (#29241)

Commit

1 year ago

Use `torch.bool` instead of `torch.int64` for non-persistant causal mask buffer (#29241) use torch.bool instead of torch.int64

References

Author

fxmarty

Parents