From 61750af66579bb90aa798918012cc9d9437a4bb1 Mon Sep 17 00:00:00 2001
From: Erik Mackdanz <erikmack@gmail.com>
Date: Sat, 30 Nov 2024 08:54:27 -0600
Subject: [PATCH] main not PR

---
 src/main.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 4a11d2b..24ccbfb 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -92,7 +92,6 @@
 //! 
 //! ## TODO
 //! 
-//! - why model from PR not main?
 //! - rename cv-*
 //! 
 use candle_transformers::models::bert::{BertModel, Config, HiddenAct, DTYPE};
@@ -201,7 +200,7 @@ struct Args {
 fn build_model_and_tokenizer(args: &Args) -> Result<(BertModel, Tokenizer)> {
     let device = candle_core::Device::Cpu;
     let default_model = "sentence-transformers/all-MiniLM-L6-v2".to_string();
-    let default_revision = "refs/pr/21".to_string();
+    let default_revision = "main".to_string();
     let (model_id, revision) = match (args.model_id.to_owned(), args.revision.to_owned()) {
         (Some(model_id), Some(revision)) => (model_id, revision),
         (Some(model_id), None) => (model_id, "main".to_string()),
-- 
2.52.0