mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-11-05 01:07:28 +00:00
[GH-ISSUE #1070] [Bug] #753
Labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/ktransformers#753
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @chenhongxx on GitHub (Apr 7, 2025).
Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/1070
检查清单
问题描述
(ktransformers) huadi@huadi:~/ktransformers$ python -m ktransformers.local_chat --model_path ./DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF
2025-04-07 21:07:29,971 - INFO - flashinfer.jit: Prebuilt kernels not found, using JIT backend
found flashinfer
Traceback (most recent call last):
File "", line 198, in _run_module_as_main
File "", line 88, in _run_code
File "/home/huadi/ktransformers/ktransformers/local_chat.py", line 26, in
from ktransformers.models.modeling_deepseek import DeepseekV2ForCausalLM
File "/home/huadi/ktransformers/ktransformers/models/modeling_deepseek.py", line 70, in
from flash_attn import flash_attn_func, flash_attn_varlen_func, flash_attn_with_kvcache
File "/home/huadi/anaconda3/envs/ktransformers/lib/python3.11/site-packages/flash_attn/init.py", line 3, in
from flash_attn.flash_attn_interface import (
File "/home/huadi/anaconda3/envs/ktransformers/lib/python3.11/site-packages/flash_attn/flash_attn_interface.py", line 15, in
import flash_attn_2_cuda as flash_attn_gpu
ImportError: /home/huadi/anaconda3/envs/ktransformers/lib/python3.11/site-packages/flash_attn_2_cuda.cpython-311-x86_64-linux-gnu.so: undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationESs
复现步骤
python -m ktransformers.local_chat --model_path ./DeepSeek-V2-Lite-Chat --gguf_path ./DeepSeek-V2-Lite-Chat-GGUF
环境信息
ubuntu22.04 4090 amd9754*2 1.1T ram cuda12.6
@qiyuxinlin commented on GitHub (Apr 7, 2025):
Hi @chenhongxx, I've encountered this bug before, and I was able to resolve it by switching the flash_attn version. You might want to give it a try.