mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
Junya Morioka
46 lines
1.0 KiB
YAML
46 lines
1.0 KiB
YAML
services:
|
|
runner:
|
|
privileged: true
|
|
restart: always
|
|
env_file:
|
|
- .env
|
|
environment:
|
|
REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
|
|
RUNNER_NAME: self-hosted-runner
|
|
RUNNER_GROUP: default
|
|
TARGET_ARCH: x64
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
args:
|
|
GH_RUNNER_VERSION: 2.329.0
|
|
TARGET_ARCH: x64
|
|
PLATFORM: linux/amd64
|
|
volumes:
|
|
- fa-self:/var/lib/docker
|
|
|
|
# runner-arm:
|
|
# platform: linux/arm64
|
|
# privileged: true
|
|
# restart: always
|
|
# env_file:
|
|
# - .env.arm
|
|
# environment:
|
|
# REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
|
|
# RUNNER_NAME: self-hosted-runner-arm
|
|
# RUNNER_GROUP: default
|
|
# TARGET_ARCH: arm64
|
|
# build:
|
|
# context: .
|
|
# dockerfile: Dockerfile
|
|
# args:
|
|
# GH_RUNNER_VERSION: 2.329.0
|
|
# TARGET_ARCH: arm64
|
|
# PLATFORM: linux/arm64
|
|
# volumes:
|
|
# - fa-self-arm:/var/lib/docker
|
|
|
|
volumes:
|
|
fa-self:
|
|
# fa-self-arm:
|