diff --git a/.github/workflows/on_push_deploy_to_staging.yaml b/.github/workflows/on_push_deploy_to_staging.yaml index db0acf3e5..1e02583fc 100644 --- a/.github/workflows/on_push_deploy_to_staging.yaml +++ b/.github/workflows/on_push_deploy_to_staging.yaml @@ -75,7 +75,7 @@ jobs: docker rm moonlite-sd || true # Run the new Docker container - docker run -d -p 7861:7861 --name moonlite-sd \ + docker run --gpus all -d -p 7861:7861 --name moonlite-sd \ --env-file /home/${{ secrets.STAGING_EC2_USER }}/.env \ ${{ secrets.STAGING_ECR_URI }}:latest