Add no-paged-attention for CI

EricLBuehler · Dec 3, 2023 · a3e2d90 · a3e2d90
1 parent 9a26979
commit a3e2d90
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -33,6 +33,8 @@ futures = "0.3.29"
 tokio = { version = "1.33.0", features = ["sync"] }
 env_logger = "0.10.1"
 tracing = "0.1.40"
+
+[target.'cfg(not(feature = "no-paged-attention"))'.dependencies]
 tch = { git = "https://github.com/EricLBuehler/tch-rs.git", branch = "copy_derive", version = "0.14.0" } # pending on LaurentMazare/tch-rs#823
 torch-sys = { git = "https://github.com/EricLBuehler/tch-rs.git", branch = "copy_derive", version = "0.14.0" } # pending on LaurentMazare/tch-rs#823