This commit is contained in:
XXXXRT666 2025-04-30 22:46:31 +01:00
parent 559dbfc711
commit 756e0e3a9f
3 changed files with 9 additions and 8 deletions

View File

@ -26,16 +26,16 @@ jobs:
matrix: matrix:
include: include:
- cuda_version: 12.4 - cuda_version: 12.4
lite: false lite: true
tag_prefix: lite-cu124 tag_prefix: lite-cu124
- cuda_version: 12.4 - cuda_version: 12.4
lite: true lite: false
tag_prefix: cu124 tag_prefix: cu124
- cuda_version: 12.8 - cuda_version: 12.8
lite: false lite: true
tag_prefix: lite-cu128 tag_prefix: lite-cu128
- cuda_version: 12.8 - cuda_version: 12.8
lite: true lite: false
tag_prefix: cu128 tag_prefix: cu128
steps: steps:

View File

@ -50,9 +50,7 @@ source "$HOME/anaconda3/etc/profile.d/conda.sh"
conda config --add channels conda-forge conda config --add channels conda-forge
conda config --set channel_priority strict conda update --all -y
conda install -n base -c conda-forge conda=25.3.1
if [ "$LITE" = "true" ]; then if [ "$LITE" = "true" ]; then
bash install.sh --device "CU${CUDA_VERSION//./}" --source HF bash install.sh --device "CU${CUDA_VERSION//./}" --source HF

View File

@ -65,8 +65,11 @@ while [[ $# -gt 0 ]]; do
esac esac
done done
TARGETPLATFORM=$(uname -m | grep -q 'x86_64' && echo "linux/amd64" || echo "linux/arm64")
docker build \ docker build \
--build-arg CUDA_VERSION=$CUDA_VERSION \ --build-arg CUDA_VERSION=$CUDA_VERSION \
--build-arg LITE=$LITE \ --build-arg LITE=$LITE \
-t "${USER}/gpt-sovits:local" \ --build-arg TARGETPLATFORM="$TARGETPLATFORM"
-t "${USER}/gpt-sovits:local" \
. .