ci: swap ort for candle on cpu

jcudit · Sep 12, 2024 · 9b9a4e3 · 9b9a4e3
1 parent 65fb21d
commit 9b9a4e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -30,7 +30,7 @@ ARG SCCACHE_GHA_ENABLED
 
 COPY --from=planner /usr/src/recipe.json recipe.json
 
-RUN cargo chef cook --release --features ort --no-default-features --recipe-path recipe.json && sccache -s
+RUN cargo chef cook --release --features candle --no-default-features --recipe-path recipe.json && sccache -s
 
 COPY backends backends
 COPY core core
@@ -40,7 +40,7 @@ COPY Cargo.lock ./
 
 FROM builder AS http-builder
 
-RUN cargo build --release --bin text-embeddings-router -F ort -F http --no-default-features && sccache -s
+RUN cargo build --release --bin text-embeddings-router -F candle -F http --no-default-features && sccache -s
 
 FROM builder AS grpc-builder
 
@@ -52,7 +52,7 @@ RUN PROTOC_ZIP=protoc-21.12-linux-x86_64.zip && \
 
 COPY proto proto
 
-RUN cargo build --release --bin text-embeddings-router -F grpc -F ort --no-default-features && sccache -s
+RUN cargo build --release --bin text-embeddings-router -F grpc -F candle --no-default-features && sccache -s
 
 FROM debian:bookworm-slim AS base