Skip to content

GitLab

Switch branch/tag
  1. 17 Jul, 2023 2 commits
  3. 08 Jul, 2023 1 commit
    • Volodymyr Kyrylov's avatar
      rotary: update cos/sin cache when switching from inference mode · 70ab266a
      Volodymyr Kyrylov authored 
      This resolves RuntimeErrors after running evaluation in inference mode:

```
  File "/home/proger/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/proger/.local/lib/python3.10/site-packages/flash_attn/modules/mha.py", line 492, in forward
    qkv = self.rotary_emb(qkv)
  File "/home/proger/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/proger/.local/lib/python3.10/site-packages/flash_attn/layers/rotary.py", line 229, in forward
    return apply_rotary_emb_qkv_(
  File "/home/proger/.local/lib/python3.10/site-packages/torch/autograd/function.py", line 506, in apply
    return super().apply(*args, **kwargs)  # type: ignore[misc]
RuntimeError: Inference tensors cannot be saved for backward. To work around you can make a clone to get a normal tensor and use it in autograd.
```
      70ab266a
  5. 04 Jul, 2023 1 commit
  7. 03 Jul, 2023 2 commits
  9. 02 Jul, 2023 1 commit
  11. 02 Jun, 2023 1 commit
  13. 30 May, 2023 2 commits
  15. 27 May, 2023 1 commit
  17. 19 May, 2023 1 commit
  19. 06 May, 2023 2 commits
  21. 05 May, 2023 1 commit
  23. 21 Apr, 2023 4 commits
  25. 19 Apr, 2023 1 commit
  27. 18 Apr, 2023 2 commits
  29. 14 Apr, 2023 1 commit
  31. 13 Apr, 2023 4 commits
  33. 12 Apr, 2023 1 commit
  35. 31 Mar, 2023 2 commits
  37. 29 Mar, 2023 2 commits
  39. 22 Mar, 2023 2 commits
  41. 15 Mar, 2023 2 commits
  43. 14 Mar, 2023 1 commit
  45. 23 Jan, 2023 1 commit
  47. 19 Jan, 2023 1 commit
  49. 18 Jan, 2023 1 commit