add qwen3-asr

src/main.rs

@@ -1,15 +1,15 @@
-use std::{net::IpAddr, str::FromStr, time::Duration};
-
-use aha::{models::WhichModel, utils::{download_model, get_default_save_dir}};
+use std::{net::IpAddr, str::FromStr};
 
+use aha::{
+    models::WhichModel,
+    utils::{download_model, get_default_save_dir},
+};
 use clap::{Args, Parser, Subcommand, ValueEnum};
-use modelscope::ModelScope;
 use rocket::{
     Config,
     data::{ByteUnit, Limits},
     routes,
 };
-use tokio::time::sleep;
 
 use crate::api::init;
 mod api;
@@ -149,6 +149,8 @@ fn get_model_id(model: WhichModel) -> &'static str {
         WhichModel::Qwen2_5vl3B => "Qwen/Qwen2.5-VL-3B-Instruct",
         WhichModel::Qwen2_5vl7B => "Qwen/Qwen2.5-VL-7B-Instruct",
         WhichModel::Qwen3_0_6B => "Qwen/Qwen3-0.6B",
+        WhichModel::Qwen3ASR0_6B => "Qwen/Qwen3-ASR-0.6B",
+        WhichModel::Qwen3ASR1_7B => "Qwen/Qwen3-ASR-1.7B",
         WhichModel::Qwen3vl2B => "Qwen/Qwen3-VL-2B-Instruct",
         WhichModel::Qwen3vl4B => "Qwen/Qwen3-VL-4B-Instruct",
         WhichModel::Qwen3vl8B => "Qwen/Qwen3-VL-8B-Instruct",
@@ -171,6 +173,8 @@ fn run_list() -> anyhow::Result<()> {
         WhichModel::Qwen2_5vl3B,
         WhichModel::Qwen2_5vl7B,
         WhichModel::Qwen3_0_6B,
+        WhichModel::Qwen3ASR0_6B,
+        WhichModel::Qwen3ASR1_7B,
         WhichModel::Qwen3vl2B,
         WhichModel::Qwen3vl4B,
         WhichModel::Qwen3vl8B,
@@ -289,6 +293,14 @@ fn run_run(args: RunArgs) -> anyhow::Result<()> {
             use aha::exec::qwen3::Qwen3Exec;
             Qwen3Exec::run(&input, output.as_deref(), &weight_path)?;
         }
+        WhichModel::Qwen3ASR0_6B => {
+            use aha::exec::qwen3_asr::Qwen3ASRExec;
+            Qwen3ASRExec::run(&input, output.as_deref(), &weight_path)?;
+        }
+        WhichModel::Qwen3ASR1_7B => {
+            use aha::exec::qwen3_asr::Qwen3ASRExec;
+            Qwen3ASRExec::run(&input, output.as_deref(), &weight_path)?;
+        }
         WhichModel::Qwen3vl2B => {
             use aha::exec::qwen3vl::Qwen3vlExec;
             Qwen3vlExec::run(&input, output.as_deref(), &weight_path)?;