remove an unused import (#960)

2024-05-23 11:12:31 -07:00 · 2024-05-23 11:12:31 -07:00 · 22339db185
commit 22339db185
parent 9c0e9ee86d
1 changed files with 0 additions and 2 deletions
--- a/flash_attn/ops/triton/cross_entropy.py
+++ b/flash_attn/ops/triton/cross_entropy.py
@ -4,8 +4,6 @@ from typing import Tuple, Optional, Union

 import torch

-from einops import rearrange
-
 import triton
 import triton.language as tl