unsloth-trainer/.woodpecker/build-containers.yaml
2024-07-13 11:32:59 -04:00

34 lines
1.1 KiB
YAML

when:
- event: manual
- event: push
steps:
- name: build_gpu
image: woodpeckerci/plugin-docker-buildx
settings:
build_args:
- "torch_cuda_arch_list=6.0 6.1 8.0 8.6+PTX"
repo: gitea.simcop2387.info/simcop2387/unsloth-base-image
auto_tag: true
tags:
- "build-${CI_PIPELINE_NUMBER}"
logins:
- registry: https://gitea.simcop2387.info/
username: simcop2387
password:
from_secret: docker_token
# - name: build_cpu
# image: woodpeckerci/plugin-docker-buildx
# settings:
# repo: gitea.simcop2387.info/simcop2387/vllm-base-cpu-image
# context: vllm-mirror/
# dockerfile: vllm-mirror/Dockerfile.cpu
# auto_tag: true
# tags:
# - "build-${CI_PIPELINE_NUMBER}"
# logins:
# - registry: https://gitea.simcop2387.info/
# username: simcop2387
# password:
# from_secret: docker_token