From 74b9696eaa4cff57d4f7097956683edf2b6b90da Mon Sep 17 00:00:00 2001
From: Ankush Malaker <43288948+AnkushMalaker@users.noreply.github.com>
Date: Thu, 18 Dec 2025 18:58:57 +0000
Subject: [PATCH] Update Docker configuration to use PYTORCH_CUDA_VERSION
 instead of CUDA_VERSION for Parakeet ASR builds

---
 .github/workflows/advanced-docker-compose-build.yml | 2 +-
 extras/asr-services/docker-compose-test.yml         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/advanced-docker-compose-build.yml b/.github/workflows/advanced-docker-compose-build.yml
index 77370495..615b2e2d 100644
--- a/.github/workflows/advanced-docker-compose-build.yml
+++ b/.github/workflows/advanced-docker-compose-build.yml
@@ -199,7 +199,7 @@ jobs:
           cd ../../extras/asr-services
           for cuda_variant in cu121 cu126 cu128; do
             echo "Building parakeet-asr-${cuda_variant}"
-            export CUDA_VERSION="${cuda_variant}"
+            export PYTORCH_CUDA_VERSION="${cuda_variant}"
             docker compose build parakeet-asr
             
             img_id=$(docker compose images -q parakeet-asr | head -n1)
diff --git a/extras/asr-services/docker-compose-test.yml b/extras/asr-services/docker-compose-test.yml
index 19fd698f..4653fb9f 100644
--- a/extras/asr-services/docker-compose-test.yml
+++ b/extras/asr-services/docker-compose-test.yml
@@ -31,7 +31,7 @@ services:
       context: .
       dockerfile: Dockerfile_Parakeet
       args:
-        CUDA_VERSION: ${CUDA_VERSION:-cu126}
+        PYTORCH_CUDA_VERSION: ${PYTORCH_CUDA_VERSION:-cu126}
     image: parakeet-asr:test
     ports:
       - "8767:8765"  # Different test port for parakeet