From 7266f46549e07751a0ccb2c61f9fab1505c1db6c Mon Sep 17 00:00:00 2001 From: Erik Mackdanz Date: Sat, 30 Nov 2024 08:13:07 -0600 Subject: [PATCH] query_opt --- src/main.rs | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) diff --git a/src/main.rs b/src/main.rs index 0c2f96d..16aceda 100644 --- a/src/main.rs +++ b/src/main.rs @@ -85,7 +85,7 @@ //! //! - why model from PR not main? //! - init-model command -//! +//! - rename cv-* //! use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE}; @@ -239,13 +239,14 @@ fn init_database(dbname: String, dbhost: String, dbuser: String, dbpassword: Str .connect(NoTls)?; let result = - client.query("SELECT 1 FROM pg_database WHERE datname=$1",&[&dbname])?; - - if result.is_empty() { - println!("creating database {}", dbname); - client.execute(&format!("create database {}",dbname),&[])?; - } else { - println!("database {} exists already", dbname); + client.query_opt("SELECT 1 FROM pg_database WHERE datname=$1",&[&dbname])?; + + match result { + Some(_) => { + println!("creating database {}", dbname); + client.execute(&format!("create database {}",dbname),&[])?; + }, + None => println!("database {} exists already", dbname), } let _ = client.close(); @@ -278,6 +279,7 @@ fn get_embeddings(input: &String, model: &BertModel, mut tokenizer: Tokenizer) - .map_err(E::msg)? .get_ids() .to_vec(); + println!("tokens {:?}",tokens); let token_ids = Tensor::new(&tokens[..], device)?.unsqueeze(0)?; let token_type_ids = token_ids.zeros_like()?; println!("Loaded and encoded {:?}", start.elapsed()); -- 2.52.0