mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
[Iluvatar] Support V1_KVCACHE_SCHEDULER and paddleocr-vl rope mode (#5555)
This commit is contained in:
@@ -428,6 +428,10 @@ class ResourceManagerV1(ResourceManager):
|
||||
grid_thw = paddle.to_tensor(grid_thw, dtype="int64")
|
||||
if current_platform.is_xpu():
|
||||
from fastdeploy.model_executor.ops.xpu import get_img_boundaries
|
||||
elif current_platform.is_iluvatar():
|
||||
from fastdeploy.model_executor.ops.iluvatar import (
|
||||
get_img_boundaries,
|
||||
)
|
||||
else:
|
||||
from fastdeploy.model_executor.ops.gpu import get_img_boundaries
|
||||
|
||||
|
||||
Reference in New Issue
Block a user