mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 08:37:06 +08:00
dcu adapter ernie45t (#2756)
Co-authored-by: lifu <lifu@sugon.com> Co-authored-by: yongqiangma <xing.wo@163.com>
This commit is contained in:
@@ -20,7 +20,7 @@ import paddle
|
||||
|
||||
from fastdeploy.platforms import current_platform
|
||||
|
||||
if current_platform.is_cuda():
|
||||
if current_platform.is_cuda() and not current_platform.is_dcu():
|
||||
from fastdeploy.model_executor.ops.gpu import \
|
||||
append_attention as append_attention_gpu
|
||||
|
||||
|
Reference in New Issue
Block a user