-
jazzcook15 authored
* Update sqrt computation so it can survive a torch.jit.trace * Update modeling_gpt2.py Co-authored-by:Patrick von Platen <patrick.v.platen@gmail.com>
c7d06b79
* Update sqrt computation so it can survive a torch.jit.trace
* Update modeling_gpt2.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>