diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile
index d786e55..335f788 100644
--- a/comfyui-rocm-dockerfile
+++ b/comfyui-rocm-dockerfile
@@ -22,7 +22,8 @@ WORKDIR /app
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
 
 # Install ComfyUI dependencies
-RUN grep -vE 'torch|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
+RUN grep -vE 'torch|torchsde|torchaudio|torchvision' requirements.txt > req_no_torch.txt \
+    && pip3 install --no-deps torchsde \
     && pip3 install --no-cache-dir -r req_no_torch.txt
 
 # Setup Manager