apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-30 11:26:39 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

[BugFix] Fix image_feature 0-Size causing insert failed (#4042)

Browse Source 
* update

* fix image_feature
This commit is contained in:
Ayakouji
2025-09-12 19:13:08 +08:00
committed by GitHub
parent 9ac539471d
commit 987609c894
2 changed files with 18 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
fastdeploy/model_executor/graph_optimization/decorator.py
View File

@@ -114,14 +114,14 @@ def cuda_graph_buffers(buffer_meta):
cur = getattr(cur, p)
return cur
if not hasattr(self, "_mm_buffers"):
self._mm_buffers = {}
if not hasattr(self, "_cuda_graph_buffers"):
self._cuda_graph_buffers = {}
for name, meta in buffer_meta.items():
shape = [_resolve_path(fd_config, s) if isinstance(s, str) else s for s in meta["shape"]]
dtype = meta["dtype"]
if "." in meta["dtype"]:
dtype = _resolve_path(fd_config, meta["dtype"])
self._mm_buffers[name] = paddle.full(
self._cuda_graph_buffers[name] = paddle.full(
shape=shape,
dtype=dtype,
fill_value=meta.get("value", 0),