diff --git a/Dockerfile b/Dockerfile
index ebe0130..054fd17 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -56,8 +56,8 @@ RUN python3 setup.py bdist_wheel
 
 # ── 编译 flashinfer (主分支支持 torch 2.7 / cu126) ─────────────────────────
 WORKDIR /opt
-RUN git clone https://github.com/Dao-AILab/flashinfer.git
-WORKDIR /opt/flashinfer/python
+RUN git clone https://github.com/flashinfer-ai/flashinfer.git
+WORKDIR /opt/flashinfer
 RUN python3 setup.py bdist_wheel
 
 # ── 编译你本地 sglang 源码并打 wheel ───────────────────────────────────────
@@ -69,8 +69,8 @@ RUN python3 -m pip install ".[srt,openai]" --no-build-isolation && \
 # ── 收集所有 wheel 到 /wheels ──────────────────────────────────────────────
 RUN mkdir -p /wheels && \
     cp /tmp/torch_dist/torch*.whl /wheels/ && \
-    cp /opt/vision/dist/torchvision-0.22.1+cu126*.whl /wheels/ && \
-    cp /opt/flashinfer/python/dist/flashinfer_python-*.whl /wheels/ && \
+    cp /opt/vision/dist/torchvision-*.whl /wheels/ && \
+    cp /opt/flashinfer/dist/flashinfer_python-*.whl /wheels/ && \
     cp /tmp/sg_wheels/sglang-*.whl /wheels/
 
 ###############################################################################