PDDecodeServer#
- class ray.serve.llm.deployment.PDDecodeServer(llm_config: LLMConfig, *, prefill_server: DeploymentHandle, engine_cls=None, model_downloader=None)[source]#
Bases:
PDDecodeServerDecode-side LLM server for prefill-decode disaggregation.
This deployment owns a real engine (decode config) and holds a handle to the prefill deployment. For chat/completions it runs remote prefill first, then local decode.
Use
build_pd_openai_appto construct the full 3-tier PD graph.Methods
Check the health of the replica.
Execute a collective RPC call on all workers.
Detokenize the input token IDs.
Runs an embeddings request to the engine and returns the response.
Check whether the engine is currently paused.
Check whether the engine is currently sleeping.
Pause generation on the engine.
Serve request-router hook, polled by the controller.
Reset the KV prefix cache on the engine.
Resume generation on the engine after pause.
Runs a score request to the engine and returns the response.
Put the engine to sleep.
Start the underlying engine.
Start profiling
Stop profiling
Synchronous constructor that returns an unstarted instance.
Tokenize the input text.
Runs an transcriptions request to the engine and returns the response.
Wake up the engine from sleep mode.