transformers
395e566a - gpt-bigcode: avoid `zero_` to support Core ML (#24755)

Commit
2 years ago
gpt-bigcode: avoid `zero_` to support Core ML (#24755) gpt-bigcode: avoid `zeros_` to support Core ML. In-place `zeros_` is not supported by the Core ML conversion process. This PR replaces it with `zeros_like` so conversion can proceed. The change only affects a workaround for a PyTorch bug on the `cpu` device.
Author
Parents
Loading