1. 23 Jul, 2023 2 commits
  2. 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
  3. 02 Jul, 2023 1 commit
  4. 30 May, 2023 1 commit
  5. 14 Mar, 2023 1 commit
  6. 30 Dec, 2022 1 commit
  7. 25 Dec, 2022 1 commit
  8. 23 Dec, 2022 1 commit
  9. 21 Dec, 2022 1 commit
  10. 17 Dec, 2022 1 commit
  11. 16 Dec, 2022 1 commit
  12. 15 Dec, 2022 1 commit
  13. 18 Nov, 2022 2 commits
  14. 17 Nov, 2022 1 commit
  15. 14 Nov, 2022 1 commit