diff --git a/src/main.rs b/src/main.rs index 7136645..e602ca3 100644 --- a/src/main.rs +++ b/src/main.rs @@ -62,7 +62,7 @@ async fn main() { description = "Bot commands. Most of the local models have Q4_K_M quantization. Some joke commands are hidden. Contact: @DuckyBlender" )] enum Commands { - #[command(description = "generate uncensored text using dolphin-llama3", alias = "u")] + #[command(description = "generate uncensored text using llama3-alpha-centauri-uncensored", alias = "u")] Uncensored, #[command(description = "generate caveman-like text", alias = "cv")] Caveman, diff --git a/src/models.rs b/src/models.rs index f226ffc..abf0cf7 100644 --- a/src/models.rs +++ b/src/models.rs @@ -5,7 +5,7 @@ use ollama_rs::{models::create::CreateModelRequest, Ollama}; #[derive(Debug, PartialEq, Sequence)] pub enum ModelType { // Ollama (text) - Uncensored, // dolphin-llama3 + Uncensored, // gurubot/llama3-alpha-centauri-uncensored Caveman, // caveman-llama3 (custom model) Racist, // duckyblender/racist-phi3 Furry, // furry-llama3 (custom model)