mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:37:53 -04:00
Junya Morioka
16 lines
371 B
YAML
16 lines
371 B
YAML
services:
|
|
runner:
|
|
privileged: true
|
|
restart: always
|
|
env_file:
|
|
- .env
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
args:
|
|
REPOSITORY_URL: https://github.com/mjun0812/flash-attention-prebuild-wheels
|
|
GH_RUNNER_VERSION: 2.329.0
|
|
RUNNER_NAME: self-hosted-runner
|
|
RUNNER_GROUP: default
|
|
TARGET_ARCH: x64
|