summaryrefslogtreecommitdiff
path: root/candle-core/examples/llama/main.rs
diff options
context:
space:
mode:
authorNicolas Patry <patry.nicolas@protonmail.com>2023-06-29 12:08:35 +0000
committerNicolas Patry <patry.nicolas@protonmail.com>2023-06-29 12:08:35 +0000
commitde48e6fd59ea23eeb1e9393bc2dc91e56ae1be09 (patch)
treee569fdd86239504f43507c2fe2efeccfdb37c418 /candle-core/examples/llama/main.rs
parent0958c588f606443312bfd33502fd658ad2c0ccb1 (diff)
downloadcandle-de48e6fd59ea23eeb1e9393bc2dc91e56ae1be09.tar.gz
candle-de48e6fd59ea23eeb1e9393bc2dc91e56ae1be09.tar.bz2
candle-de48e6fd59ea23eeb1e9393bc2dc91e56ae1be09.zip
Putting back main.
Diffstat (limited to 'candle-core/examples/llama/main.rs')
-rw-r--r--candle-core/examples/llama/main.rs5
1 files changed, 2 insertions, 3 deletions
diff --git a/candle-core/examples/llama/main.rs b/candle-core/examples/llama/main.rs
index c91537d8..cc8a8ae9 100644
--- a/candle-core/examples/llama/main.rs
+++ b/candle-core/examples/llama/main.rs
@@ -1,7 +1,7 @@
-// An implementation of LLaMA https://github.com/facebookresearch/llama");");");
+// An implementation of LLaMA https://github.com/facebookresearch/llama.py
//
// This is based on nanoGPT in a similar way to:
-// https://github.com/Lightning-AI/lit-llama/blob/main/lit_llama/model.py");
+// https://github.com/Lightning-AI/lit-llama/blob/main/lit_llama/model.py
//
// The tokenizer config can be retrieved from:
// https://huggingface.co/hf-internal-testing/llama-tokenizer/raw/main/tokenizer.json
@@ -417,7 +417,6 @@ struct Args {
#[tokio::main]
async fn main() -> Result<()> {
- //use rand::prelude::*;
use tokenizers::Tokenizer;
let args = Args::parse();