apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2026-04-23 17:11:21 +08:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Rename top_p_sampling to top_k_top_p_sampling (#2791)

Browse Source
This commit is contained in:
Sunny-bot1
2025-07-10 15:09:25 +08:00
committed by GitHub
parent e45050cae3
commit 1e2319cbef
5 changed files with 23 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastdeploy/model_executor/layers/sample/ops/__init__.py
+2 -2
View File
@@ -16,10 +16,10 @@
from .apply_penalty_multi_scores import (
apply_penalty_multi_scores, apply_speculative_penalty_multi_scores)
from .top_p_sampling import top_p_sampling
from .top_k_top_p_sampling import top_k_top_p_sampling
__all__ = [
"apply_penalty_multi_scores",
"apply_speculative_penalty_multi_scores",
"top_p_sampling",
"top_k_top_p_sampling",
]
fastdeploy/model_executor/layers/sample/ops/top_p_sampling.py → fastdeploy/model_executor/layers/sample/ops/top_k_top_p_sampling.py
+1 -1
View File
@@ -25,7 +25,7 @@ if current_platform.is_gcu():
from fastdeploy.model_executor.ops.gcu import \
top_p_sampling as gcu_top_p_sampling
def top_p_sampling(
def top_k_top_p_sampling(
x: paddle.Tensor,
top_p: paddle.Tensor,
top_k: Optional[paddle.Tensor] = None,