geoffsee/predict-otron-9001
1
0
Fork 0
mirror of https://github.com/geoffsee/predict-otron-9001.git synced 2025-09-08 22:46:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fmt and clippy

Browse Source
This commit is contained in:
geoffsee
2025-09-04 15:07:49 -04:00
parent 3ecdd9ffa0
commit 3992532f15
13 changed files with 28 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
integration/gemma-runner/src/gemma_api.rs
View File

@@ -205,7 +205,7 @@ impl TextGeneration {
let start_gen = std::time::Instant::now();
for (_generated_tokens, index) in (0..sample_len).enumerate() {
for index in 0..sample_len {
let context_size = if index > 0 { 1 } else { tokens.len() };
let start_pos = tokens.len().saturating_sub(context_size);
let ctxt = &tokens[start_pos..];

2
integration/gemma-runner/src/main.rs
View File

@@ -6,10 +6,8 @@ mod gemma_api;
mod gemma_cli;
use anyhow::Error;
use clap::{Parser, ValueEnum};
use crate::gemma_cli::run_cli;
use std::io::Write;
/// just a placeholder, not used for anything
fn main() -> std::result::Result<(), Error> {