github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove duplicate import (#5100)

pull/5114/head
Xuanlei Zhao 2023-11-23 15:15:01 +08:00 committed by GitHub
parent e53e729d8e
commit 68fcaa2225
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/shardformer/modeling/llama.py
View File

@ -413,8 +413,6 @@ def get_llama_flash_attention_forward():
warnings.warn("using llamav1, llamav1 hasn't repeat_kv function") warnings.warn("using llamav1, llamav1 hasn't repeat_kv function")
llama_version = 1 llama_version = 1
from colossalai.kernel.cuda_native import AttnMaskType, ColoAttention
def forward( def forward(
self: LlamaAttention, self: LlamaAttention,
hidden_states: torch.Tensor, hidden_states: torch.Tensor,