From 94aa0818b36202f925ce3fb89d5a5c5e895d97e3 Mon Sep 17 00:00:00 2001
From: Andrew Beveridge <andrew.beveridge@transferwise.com>
Date: Sat, 30 Dec 2023 19:40:41 -0600
Subject: [PATCH] Attempt to fix CI build for runpod image

---
 .github/workflows/publish-to-conda.yml  | 2 +-
 .github/workflows/publish-to-docker.yml | 4 ++--
 pyproject.toml                          | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/workflows/publish-to-conda.yml b/.github/workflows/publish-to-conda.yml
index 36b7009..0b1011a 100644
--- a/.github/workflows/publish-to-conda.yml
+++ b/.github/workflows/publish-to-conda.yml
@@ -24,7 +24,7 @@ jobs:
         with:
           python-version: ${{ matrix.python-version }}
           environment-file: conda/build_env.yaml
-          channels: conda-forge
+          channels: conda-forge,pytorch
           mamba-version: "*"
           use-mamba: true
           auto-update-conda: false
diff --git a/.github/workflows/publish-to-docker.yml b/.github/workflows/publish-to-docker.yml
index b854d4c..af2bc4e 100644
--- a/.github/workflows/publish-to-docker.yml
+++ b/.github/workflows/publish-to-docker.yml
@@ -57,7 +57,7 @@ jobs:
         with:
           file: Dockerfile.gpu
           context: .
-          platforms: linux/amd64,linux/arm64
+          platforms: linux/amd64
           push: true
           tags: |
             beveradb/audio-separator:gpu-${{ env.VERSION }}
@@ -69,7 +69,7 @@ jobs:
         with:
           file: Dockerfile.runpod
           context: .
-          platforms: linux/amd64,linux/arm64
+          platforms: linux/amd64
           push: true
           tags: |
             beveradb/audio-separator:runpod-${{ env.VERSION }}
diff --git a/pyproject.toml b/pyproject.toml
index e6c199f..3bb0c23 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "audio-separator"
-version = "0.11.2"
+version = "0.11.3"
 description = "Easy to use vocal separation, using MDX-Net models from UVR trained by @Anjok07"
 authors = ["Andrew Beveridge <andrew@beveridge.uk>"]
 license = "MIT"