update fmt
This commit is contained in:
+2
-2
@@ -34,8 +34,8 @@ fn qwen3_0_6b_generate() -> Result<()> {
|
||||
let result = model.generate(mes)?;
|
||||
let i_duration = i_start.elapsed();
|
||||
println!("generate: \n {:?}", result);
|
||||
if result.usage.is_some() {
|
||||
let num_token = result.usage.as_ref().unwrap().total_tokens;
|
||||
if let Some(usage) = &result.usage {
|
||||
let num_token = usage.total_tokens;
|
||||
let duration_secs = i_duration.as_secs_f64();
|
||||
let tps = num_token as f64 / duration_secs;
|
||||
println!("Tokens per second (TPS): {:.2}", tps);
|
||||
|
||||
Reference in New Issue
Block a user