mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 23:07:09 +08:00
13 lines
298 B
Python
13 lines
298 B
Python
from deepinfer.npu_ops.patch_imports import patch_imports
|
|
|
|
patch_imports(
|
|
{
|
|
"deepinfer.npu_ops.deep_select": {
|
|
"deep_select.interface": "topk",
|
|
},
|
|
"deepinfer.npu_ops.vit_attn": {
|
|
"flash_attn.cute.interface": "_flash_attn_fwd",
|
|
},
|
|
}
|
|
)
|