Skip to content

Commit

Permalink
first commit (#54)
Browse files Browse the repository at this point in the history
  • Loading branch information
jorgeantonio21 authored Apr 25, 2024
1 parent 1c3eb39 commit 25a4479
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions atoma-inference/src/models/candle/llama.rs
Original file line number Diff line number Diff line change
Expand Up @@ -181,14 +181,12 @@ impl ModelTrait for LlamaModel {
}
if let Some(t) = tokenizer.next_token(next_token)? {
res += &t;
info!("res: {res}");
}

generated_tokens += 1;
}
if let Some(rest) = tokenizer.decode_rest()? {
res += &rest;
info!("res: {res}");
}

let dt = start_gen.elapsed();
Expand Down

0 comments on commit 25a4479

Please sign in to comment.