Skip to content

Commit

Permalink
Commented out some tests for the moment
Browse files Browse the repository at this point in the history
  • Loading branch information
cpetersen committed Mar 13, 2024
1 parent 5288301 commit f1585a8
Showing 1 changed file with 21 additions and 21 deletions.
42 changes: 21 additions & 21 deletions ext/candle/src/model/config.rs
Original file line number Diff line number Diff line change
Expand Up @@ -134,24 +134,24 @@ impl ModelConfig {
}
}

#[cfg(test)]
mod tests {
#[test]
fn my_first_test() {
assert_eq!(2 + 2, 4);
}

#[test]
fn test_build_model_and_tokenizer() {
let config = super::ModelConfig::build();
let (_model, tokenizer) = config.build_model_and_tokenizer().unwrap();
assert_eq!(tokenizer.get_vocab_size(true), 30522);
}

#[test]
fn test_embedding() {
let config = super::ModelConfig::build();
// let (_model, tokenizer) = config.build_model_and_tokenizer().unwrap();
// assert_eq!(config.embedding("Scientist.com is a marketplace for pharmaceutical services.")?, None);
}
}
// #[cfg(test)]
// mod tests {
// #[test]
// fn my_first_test() {
// assert_eq!(2 + 2, 4);
// }

// #[test]
// fn test_build_model_and_tokenizer() {
// let config = super::ModelConfig::build();
// let (_model, tokenizer) = config.build_model_and_tokenizer().unwrap();
// assert_eq!(tokenizer.get_vocab_size(true), 30522);
// }

// #[test]
// fn test_embedding() {
// let config = super::ModelConfig::build();
// // let (_model, tokenizer) = config.build_model_and_tokenizer().unwrap();
// // assert_eq!(config.embedding("Scientist.com is a marketplace for pharmaceutical services.")?, None);
// }
// }

0 comments on commit f1585a8

Please sign in to comment.