- 23 Jul, 2023 2 commits
- 08 Jul, 2023 1 commit
-
-
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. ```
-
- 02 Jul, 2023 1 commit
-
-
Tri Dao authored
-
- 30 May, 2023 1 commit
-
-
Tri Dao authored
-
- 14 Mar, 2023 1 commit
-
-
Tri Dao authored
-
- 30 Dec, 2022 1 commit
-
-
Tri Dao authored
-
- 25 Dec, 2022 1 commit
-
-
Tri Dao authored
-
- 23 Dec, 2022 1 commit
-
-
Tri Dao authored
-
- 21 Dec, 2022 1 commit
-
-
Tri Dao authored
-
- 17 Dec, 2022 1 commit
-
-
Alexander Ploshkin authored
-
- 16 Dec, 2022 1 commit
-
-
Alexander Ploshkin authored
-
- 15 Dec, 2022 1 commit
-
-
Alexander Ploshkin authored
-
- 18 Nov, 2022 2 commits
- 17 Nov, 2022 1 commit
-
-
Tri Dao authored
-
- 14 Nov, 2022 1 commit
-
-
Tri Dao authored
-