mirror of
https://github.com/hanzo-ml/kserve.git
synced 2026-10-30 20:53:49 +00:00
udaij12andGitHub
4ea9f747e5
* security update Signed-off-by: udai <[email protected]> * adding sign off Signed-off-by: udai <[email protected]> --------- Signed-off-by: udai <[email protected]>
17 lines
371 B
YAML
17 lines
371 B
YAML
apiVersion: "serving.kserve.io/v1beta1"
|
|
kind: "InferenceService"
|
|
metadata:
|
|
name: "torchserve-gpu"
|
|
spec:
|
|
predictor:
|
|
pytorch:
|
|
storageUri: gs://kfserving-examples/models/torchserve/image_classifier/v1
|
|
resources:
|
|
limits:
|
|
memory: 4Gi
|
|
nvidia.com/gpu: "1"
|
|
args:
|
|
- --disable-token-auth
|
|
- --enable-model-api
|
|
|