mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
add cache queue port (#3904) (#3926)
Some checks failed
CE Compile Job / ce_job_pre_check (push) Has been cancelled
CE Compile Job / print_ce_job_pre_check_outputs (push) Has been cancelled
CE Compile Job / FD-Clone-Linux (push) Has been cancelled
CE Compile Job / Show Code Archive Output (push) Has been cancelled
CE Compile Job / BUILD_SM8090 (push) Has been cancelled
CE Compile Job / BUILD_SM8689 (push) Has been cancelled
CE Compile Job / CE_UPLOAD (push) Has been cancelled
Some checks failed
CE Compile Job / ce_job_pre_check (push) Has been cancelled
CE Compile Job / print_ce_job_pre_check_outputs (push) Has been cancelled
CE Compile Job / FD-Clone-Linux (push) Has been cancelled
CE Compile Job / Show Code Archive Output (push) Has been cancelled
CE Compile Job / BUILD_SM8090 (push) Has been cancelled
CE Compile Job / BUILD_SM8689 (push) Has been cancelled
CE Compile Job / CE_UPLOAD (push) Has been cancelled
* add cache queue port * add cache queue port * add cache queue port
This commit is contained in:
@@ -78,6 +78,7 @@ def form_model_get_output_topp0(
|
||||
load_choices,
|
||||
engine_worker_queue_port,
|
||||
prompts,
|
||||
cache_queue_port,
|
||||
result_queue,
|
||||
):
|
||||
try:
|
||||
@@ -88,6 +89,7 @@ def form_model_get_output_topp0(
|
||||
load_choices=load_choices,
|
||||
quantization=quantization,
|
||||
engine_worker_queue_port=engine_worker_queue_port,
|
||||
cache_queue_port=cache_queue_port,
|
||||
) as fd_model:
|
||||
fd_outputs = fd_model.generate_topp0(prompts, max_tokens=max_tokens)
|
||||
result_queue.put(fd_outputs)
|
||||
|
||||
Reference in New Issue
Block a user