Adding Github actions for building the docker images

docker/base/build-base-env.sh

@@ -43,5 +43,5 @@ docker buildx \
   --progress=plain \
   --build-arg CUDA_VERSION=$CUDA_VERSION \
   --build-arg PYTHON_VERSION=$PYTHON_VERSION \
-  -t ${GITHUB_REPOSITORY}alltalk_environment:${DOCKER_TAG} \
+  -t ${GITHUB_REPOSITORY}alltalk_tts_environment:${DOCKER_TAG} \
   .

docker/deepspeed/Dockerfile

@@ -1,5 +1,6 @@
 ARG GITHUB_REPOSITORY
-FROM ${GITHUB_REPOSITORY}alltalk_environment:latest
+ARG DOCKER_TAG=latest
+FROM ${GITHUB_REPOSITORY}alltalk_tts_environment:${DOCKER_TAG}
 
 ARG DEEPSPEED_VERSION=0.16.3
 ENV DEEPSPEED_VERSION=$DEEPSPEED_VERSION

docker/deepspeed/build-deepspeed.sh

@@ -61,5 +61,6 @@ docker run \
   -it \
   --gpus=all \
   --name deepspeed \
+  --build-arg DOCKER_TAG=$DOCKER_TAG \
   -v $SCRIPT_DIR/build:/deepspeed \
   ${GITHUB_REPOSITORY}alltalk_deepspeed:${DOCKER_TAG}

docker/variables.sh

@@ -1,7 +1,17 @@
 #!/usr/bin/env bash
 
-export CUDA_VERSION=12.6.0
-export PYTHON_VERSION=3.11.11
-export DEEPSPEED_VERSION=0.16.2
-export ALLTALK_DIR=/opt/alltalk
-export GITHUB_REPOSITORY=
+declare -A ALLTALK_VARS
+ALLTALK_VARS["CUDA_VERSION"]=12.6.0
+ALLTALK_VARS["PYTHON_VERSION"]=3.11.11
+ALLTALK_VARS["DEEPSPEED_VERSION"]=0.16.2
+ALLTALK_VARS["ALLTALK_DIR"]=/opt/alltalk
+ALLTALK_VARS["GITHUB_REPOSITORY"]=
+
+# Export single variables (needed by Docker locally)
+for key in "${!ALLTALK_VARS[@]}"
+do
+  export "${key}=${ALLTALK_VARS[${key}]}"
+done
+
+# Export the entire associative array (needed by Github action)
+export ALLTALK_VARS