diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 87e78ac..98c7bba 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -11,6 +11,8 @@ jobs: config: - { docker-file: 'Dockerfile.cpu', image-name: 'facefusion-cpu' } - { docker-file: 'Dockerfile.cuda', image-name: 'facefusion-cuda' } + - { docker-file: 'Dockerfile.tensorrt', image-name: 'facefusion-tensorrt' } + - { docker-file: 'Dockerfile.rocm', image-name: 'facefusion-rocm' } steps: - name: Checkout uses: actions/checkout@v2