summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorLaurent Mazare <laurent.mazare@gmail.com>2025-01-13 18:04:14 +0100
committerGitHub <noreply@github.com>2025-01-13 18:04:14 +0100
commit158817f230095f4a3599a29c30c0a3ae48c10b01 (patch)
tree15854d3bc275ac1acd9858699984cca825f0e19f
parent309cd0f7c7d2035f3f43da8a4cd7e6a7a897c515 (diff)
downloadcandle-158817f230095f4a3599a29c30c0a3ae48c10b01.tar.gz
candle-158817f230095f4a3599a29c30c0a3ae48c10b01.tar.bz2
candle-158817f230095f4a3599a29c30c0a3ae48c10b01.zip
Helium repo update. (#2716)
-rw-r--r--candle-examples/examples/helium/README.md8
-rw-r--r--candle-examples/examples/helium/main.rs2
2 files changed, 8 insertions, 2 deletions
diff --git a/candle-examples/examples/helium/README.md b/candle-examples/examples/helium/README.md
index 9d1f2009..2befd101 100644
--- a/candle-examples/examples/helium/README.md
+++ b/candle-examples/examples/helium/README.md
@@ -1,6 +1,12 @@
# candle-helium: 2b LLM with CC-BY licensed weights
-- [Model card](https://huggingface.co/kyutai/helium-1-preview) on the HuggingFace Hub.
+Helium-1 is a lightweight model with around 2B parameters, the preview version
+currently supports 6 languages, showing strong capabilities in those languages
+compared to existing open weights models.
+
+- [Blog Post](https://kyutai.org/2025/01/13/helium.html) announcing the model
+ release.
+- [Model card](https://huggingface.co/kyutai/helium-1-preview-2b) on the HuggingFace Hub.
## Running the example
diff --git a/candle-examples/examples/helium/main.rs b/candle-examples/examples/helium/main.rs
index d427f104..8cf63758 100644
--- a/candle-examples/examples/helium/main.rs
+++ b/candle-examples/examples/helium/main.rs
@@ -229,7 +229,7 @@ fn main() -> Result<()> {
Some(model_id) => model_id,
None => {
let name = match args.which {
- Which::V1Preview => "kyutai/helium-1-preview",
+ Which::V1Preview => "kyutai/helium-1-preview-2b",
};
name.to_string()
}