PromptCompression/cloudbuild.gpu.yaml at main · FocusedObjective/PromptCompression · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
steps:
  - name: gcr.io/cloud-builders/docker
    args:
      - build
      - -f
      - Dockerfile.gpu
      - --build-arg
      - GPU_BASE_IMAGE=${_GPU_BASE_IMAGE}
      - --build-arg
      - COMPRESSOR_MODEL=${_COMPRESSOR_MODEL}
      - --build-arg
      - ENABLE_ONNX_RUNTIME=${_ENABLE_ONNX_RUNTIME}
      - -t
      - ${_REGION}-docker.pkg.dev/$PROJECT_ID/${_REPO}/${_IMAGE_NAME}:${_IMAGE_TAG}
      - .

images:
  - ${_REGION}-docker.pkg.dev/$PROJECT_ID/${_REPO}/${_IMAGE_NAME}:${_IMAGE_TAG}

substitutions:
  _REGION: us-central1
  _REPO: prompt-compression
  _IMAGE_NAME: prompt-compression-gpu
  _IMAGE_TAG: latest
  _GPU_BASE_IMAGE: us-docker.pkg.dev/deeplearning-platform-release/gcr.io/pytorch-cu124.2-4.py310
  _COMPRESSOR_MODEL: microsoft/llmlingua-2-bert-base-multilingual-cased-meetingbank
  _ENABLE_ONNX_RUNTIME: "false"

options:
  machineType: E2_HIGHCPU_32
  diskSizeGb: "500"