Pin flashinfer

predibase · Aug 21, 2024 · 53cd493 · 53cd493
1 parent 1095b52
commit 53cd493
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -219,7 +219,7 @@ COPY --from=eetq-kernels-builder /usr/src/eetq/build/lib.linux-x86_64-cpython-31
 RUN pip install einops --no-cache-dir
 
 # Install flashinfer
-RUN pip install flashinfer -i https://flashinfer.ai/whl/cu124/torch2.4
+RUN pip install flashinfer==0.1.5+cu124torch2.4 -i https://flashinfer.ai/whl/cu124/torch2.4
 
 # Install server
 COPY proto proto