diff --git a/Cargo.toml b/Cargo.toml index 85720b9..be5e37b 100644 --- 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