diff --git a/docker-build.sh b/docker-build.sh
index 0d52e0a..c61f089 100755
--- a/docker-build.sh
+++ b/docker-build.sh
@@ -9,6 +9,11 @@ TTS_MODEL=xtts
 DOCKER_TAG=latest
 CLEAN=false
 
+# Create required build directories if they don't exist
+mkdir -p ${SCRIPT_DIR=}/docker/conda/build
+mkdir -p ${SCRIPT_DIR=}/docker/deepspeed/build
+echo "Created build directories"
+
 # Parse arguments
 while [ "$#" -gt 0 ]; do
   case "$1" in
@@ -42,6 +47,10 @@ done
 if [ "$CLEAN" = true ]; then
   rm -rf ${SCRIPT_DIR=}/docker/conda/build
   rm -rf ${SCRIPT_DIR=}/docker/deepspeed/build
+  # Recreate directories after clean
+  mkdir -p ${SCRIPT_DIR=}/docker/conda/build
+  mkdir -p ${SCRIPT_DIR=}/docker/deepspeed/build
+  echo "Cleaned and recreated build directories"  
 fi
 
 $SCRIPT_DIR/docker/conda/build-conda-env.sh \
@@ -62,4 +71,4 @@ docker buildx \
   -t alltalk_beta:${DOCKER_TAG} \
   .
 
-echo "Docker build process finished. Use docker-start.sh to start the container."
\ No newline at end of file
+echo "Docker build process finished. Use docker-start.sh to start the container."