From 917b15f35ecb99566ea5384598846da7fc3211d5 Mon Sep 17 00:00:00 2001
From: Derek Holloway <derek@mistox.com>
Date: Mon, 27 Apr 2026 18:06:23 -0700
Subject: [PATCH] Fix paths

---
 comfyui-rocm-dockerfile | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index 4006dc8..4da03f9 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -36,6 +36,8 @@ RUN git clone --depth 1 --branch v2.2.2 https://github.com/pytorch/audio.git &&
     cd audio && \
     git submodule update --init --recursive && \
     export PIP_BREAK_SYSTEM_PACKAGES=1 && \
+    export Torch_DIR=/usr/lib/python3/dist-packages/torch/share/cmake/Torch && \
+    export CMAKE_PREFIX_PATH=/usr/lib/python3/dist-packages/torch/share/cmake && \
     USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
     cd .. && rm -rf audio