From 22339db185027324f334a7f59e2584da266bfd4c Mon Sep 17 00:00:00 2001 From: lancerts Date: Thu, 23 May 2024 11:12:31 -0700 Subject: [PATCH] remove an unused import (#960) --- flash_attn/ops/triton/cross_entropy.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/flash_attn/ops/triton/cross_entropy.py b/flash_attn/ops/triton/cross_entropy.py index 1f895d7..1782338 100644 --- 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