diff options
Diffstat (limited to 'candle-flash-attn/Cargo.toml')
| -rw-r--r-- | candle-flash-attn/Cargo.toml | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/candle-flash-attn/Cargo.toml b/candle-flash-attn/Cargo.toml index 25201a0e..9d21cf4a 100644 --- a/candle-flash-attn/Cargo.toml +++ b/candle-flash-attn/Cargo.toml @@ -16,3 +16,5 @@ half = { version = "2.3.1", features = ["num-traits"] } [build-dependencies] anyhow = { version = "1", features = ["backtrace"] } +num_cpus = "1.15.0" +rayon = "1.7.0" |