Skip to content

Commit b3f3803

Browse files
authored
doc: Update doc for model_server_workers param in PyTorchModel (#3868)
1 parent 0ecb389 commit b3f3803

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/pytorch/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -123,9 +123,9 @@ def __init__(
123123
to call to create a predictor with an endpoint name and
124124
SageMaker ``Session``. If specified, ``deploy()`` returns the
125125
result of invoking this function on the created endpoint name.
126-
model_server_workers (int or PipelineVariable): Optional. The number of worker processes
127-
used by the inference server. If None, server will use one
128-
worker per vCPU.
126+
model_server_workers (int or PipelineVariable): Optional. The number of worker
127+
processes used by the TorchServe model server. If None, available GPUs
128+
in system or number of logical processors available to the JVM.
129129
**kwargs: Keyword arguments passed to the superclass
130130
:class:`~sagemaker.model.FrameworkModel` and, subsequently, its
131131
superclass :class:`~sagemaker.model.Model`.

0 commit comments

Comments
 (0)