update WhichModel enum

tests/messy_test.rs

@@ -5,7 +5,10 @@
 // use std::io::{Read, Seek};
 // use std::{io::Cursor, time::Instant};
 
-use aha::utils::{timestamp, timestamp_millis};
+use aha::{
+    models::common::model_mapping::WhichModel,
+    utils::{timestamp, timestamp_millis},
+};
 // use aha::utils::tensor_utils::repeat_interleave;
 // use crate::params::chat::ChatCompletionParameters;
 use anyhow::Result;
@@ -36,8 +39,12 @@ async fn download_test() -> Result<()> {
 #[test]
 fn messy_test() -> Result<()> {
     // RUST_BACKTRACE=1 cargo test -F cuda --test messy_test messy_test -r -- --nocapture
-    println!("当前秒级时间戳: {}", timestamp());
-    println!("当前毫秒级时间戳: {}", timestamp_millis());
+    let model = WhichModel::LFM2_1_2B;
+    println!("model: {:?}, model_id: {}", model, model.as_string());
+    let model_list = WhichModel::model_list();
+    println!("model_list: {:#?}", model_list);
+    // println!("当前秒级时间戳: {}", timestamp());
+    // println!("当前毫秒级时间戳: {}", timestamp_millis());
 
     // let t1 = Tensor::randn(0.0, 1.0, (1, 2, 6), device)?;
     // println!(" t1: {}", t1);