Skip to content

Commit 1540065

Browse files
committed
perf(embeddings): enable flush-to-zero for onnx
1 parent 85ade44 commit 1540065

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

embeddings/src/model/local.rs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -810,6 +810,8 @@ impl OnnxEmbeddingModel {
810810
.map_err(|_| LibError::OnnxModelEvalFailed)?
811811
.with_intra_op_spinning(false)
812812
.map_err(|_| LibError::OnnxModelEvalFailed)?
813+
.with_flush_to_zero()
814+
.map_err(|_| LibError::OnnxModelEvalFailed)?
813815
.with_approximate_gelu()
814816
.map_err(|_| LibError::OnnxModelEvalFailed)?
815817
.commit_from_file(&onnx_path)

0 commit comments

Comments
 (0)