diff --git a/src/rllama_main.rs b/src/rllama_main.rs index d1328be..7a33760 100644 --- a/src/rllama_main.rs +++ b/src/rllama_main.rs @@ -22,6 +22,9 @@ struct Cli { #[arg(long)] param_path: String, + #[arg(short, long, action)] + quiet: bool, + #[arg(long)] prompt: Option, #[arg(long)] @@ -81,6 +84,12 @@ pub fn main() -> Result<(), Box> { if !colored::control::SHOULD_COLORIZE.should_colorize() { be_quiet = true; } + if cli.quiet { + be_quiet = true; + } + if be_quiet { + colored::control::SHOULD_COLORIZE.set_override(false); + } #[cfg(feature = "opencl")] let opencl: Option = {