From 4c0a33841e17f439c81bcb8e1502cb1a2027c6be Mon Sep 17 00:00:00 2001 From: laurent Date: Mon, 13 Jan 2025 18:02:33 +0100 Subject: [PATCH] Helium repo update. --- candle-examples/examples/helium/README.md | 8 +++++++- candle-examples/examples/helium/main.rs | 2 +- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/candle-examples/examples/helium/README.md b/candle-examples/examples/helium/README.md index 9d1f2009e..2befd1012 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 d427f104a..8cf63758c 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() }