summaryrefslogtreecommitdiff
path: root/candle-flash-attn/Cargo.toml
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2024-02-27 14:19:59 +0100
committerGitHub <noreply@github.com>2024-02-27 14:19:59 +0100
commit5e526abc8c0ecad2bd110a34d128e1e6d5333c68 (patch)
tree6420a6b8d0a88f87fef14399b81cf877c33945d7 /candle-flash-attn/Cargo.toml
parent6400e1b0a08d594e1448d522a41bddc98c584313 (diff)
downloadcandle-5e526abc8c0ecad2bd110a34d128e1e6d5333c68.tar.gz
candle-5e526abc8c0ecad2bd110a34d128e1e6d5333c68.tar.bz2
candle-5e526abc8c0ecad2bd110a34d128e1e6d5333c68.zip
Bump the version number to 0.4.1. (#1768)
* Fix the block size for some cuda kernels. * Bump the version number to 0.4.1.
Diffstat (limited to 'candle-flash-attn/Cargo.toml')
-rw-r--r--candle-flash-attn/Cargo.toml4
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-flash-attn/Cargo.toml b/candle-flash-attn/Cargo.toml
index bee6bbf2..29d72cd7 100644
--- a/candle-flash-attn/Cargo.toml
+++ b/candle-flash-attn/Cargo.toml
@@ -1,6 +1,6 @@
[package]
name = "candle-flash-attn"
-version = "0.4.0"
+version = "0.4.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.4.0" }
+candle = { path = "../candle-core", features = ["cuda"], package = "candle-core", version = "0.4.1" }
half = { version = "2.3.1", features = ["num-traits"] }
[build-dependencies]