apiVersion: serving.kserve.io/v1beta1
kind: InferenceService
metadata:
  name: torch-transformer
spec:
  predictor:
    triton:
      storageUri: gs://kfserving-examples/models/torchscript
      runtimeVersion: 20.10-py3
      ports:
        - name: h2c
          protocol: TCP
          containerPort: 9000
  transformer:
    containers:
      - image: kserve/image-transformer:latest
        name: kserve-container
        args:
          - --model_name
          - cifar10
          - --protocol
          - grpc-v2
