Skip to content

SD3 and Flux support #3465

SD3 and Flux support

SD3 and Flux support #3465

name: ONNX Runtime / Test ORTTrainer
on:
workflow_dispatch:
schedule:
- cron: 0 1 */3 * * # at 1am every 3 days
pull_request:
types: [opened, synchronize, reopened, labeled]
jobs:
do-the-job:
if: ${{ (github.event_name == 'workflow_dispatch') || (github.event_name == 'schedule') || contains( github.event.pull_request.labels.*.name, 'training')}}
name: Run ORTTrainer test
runs-on:
group: aws-g6-4xlarge-plus
env:
AWS_REGION: us-east-1
steps:
- name: Checkout
uses: actions/checkout@v2
- name: Build image
run: |
docker build -f tests/onnxruntime/docker/Dockerfile_onnxruntime_trainer -t onnxruntime/train .
- name: Run test within docker container
run: |
docker run --rm --gpus all -v /mnt/cache/.cache/huggingface:/root/.cache/huggingface --workdir=/workspace/optimum/tests onnxruntime/train:latest