Skip to content

GitLab

Switch branch/tag
  1. 18 Jul, 2023 1 commit
  3. 17 Jul, 2023 2 commits
  5. 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
  7. 04 Jul, 2023 1 commit
  9. 03 Jul, 2023 2 commits
  11. 02 Jul, 2023 1 commit
  13. 02 Jun, 2023 1 commit
  15. 30 May, 2023 2 commits
  17. 27 May, 2023 1 commit
  19. 19 May, 2023 1 commit
  21. 06 May, 2023 2 commits
  23. 05 May, 2023 1 commit
  25. 21 Apr, 2023 4 commits
  27. 19 Apr, 2023 1 commit
  29. 18 Apr, 2023 2 commits
  31. 14 Apr, 2023 1 commit
  33. 13 Apr, 2023 4 commits
  35. 12 Apr, 2023 1 commit
  37. 31 Mar, 2023 2 commits
  39. 29 Mar, 2023 2 commits
  41. 22 Mar, 2023 2 commits
  43. 15 Mar, 2023 2 commits
  45. 14 Mar, 2023 1 commit
  47. 23 Jan, 2023 1 commit
  49. 19 Jan, 2023 1 commit