github.com/pachyderm/pachyderm@v1.13.4/examples/ml/gpt-2/generate.json (about)

     1  {
     2      "pipeline": {
     3          "name": "generate"
     4      },
     5      "description": "A pipeline that generates tweets based on the trained model.",
     6      "transform": {
     7          "image": "pachyderm/gpt-2-example",
     8          "cmd": ["/generate.py"]
     9      },
    10      "input": {
    11          "pfs": {
    12              "repo": "train",
    13              "glob": "/*"
    14          }
    15      },
    16      "resource_limits": {
    17          "gpu": {
    18              "type": "nvidia.com/gpu",
    19              "number": 1
    20          },
    21          "memory": "10G",
    22          "cpu": 1
    23      },
    24      "resource_requests": {
    25          "memory": "10G",
    26          "cpu": 1
    27      },
    28      "standby": true
    29  }