diff options
Diffstat (limited to 'candle-flash-attn')
| -rw-r--r-- | candle-flash-attn/Cargo.toml | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-flash-attn/Cargo.toml b/candle-flash-attn/Cargo.toml index 827cf970..7574b76b 100644 --- a/candle-flash-attn/Cargo.toml +++ b/candle-flash-attn/Cargo.toml @@ -1,6 +1,6 @@ [package] name = "candle-flash-attn" -version = "0.5.0" +version = "0.5.1" edition = "2021" description = "Flash attention layer for the candle ML framework." @@ -11,7 +11,7 @@ license = "MIT OR Apache-2.0" readme = "README.md" [dependencies] -candle = { path = "../candle-core", features = ["cuda"], package = "candle-core", version = "0.5.0" } +candle = { path = "../candle-core", features = ["cuda"], package = "candle-core", version = "0.5.1" } half = { version = "2.3.1", features = ["num-traits"] } [build-dependencies] |