From 2dbeae0987f702cae85274e296b087d609fcd289 Mon Sep 17 00:00:00 2001
From: Ariver <shanghai3168@gmail.com>
Date: Wed, 03 Jun 2026 17:27:41 +0800
Subject: [PATCH] Add Qwen3-ASR Chinese advanced model
---
privatevoice.src/internal/engine/engine_darwin_test.go | 47 +++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 47 insertions(+), 0 deletions(-)
diff --git a/privatevoice.src/internal/engine/engine_darwin_test.go b/privatevoice.src/internal/engine/engine_darwin_test.go
index d4d00d1..8310a03 100644
--- a/privatevoice.src/internal/engine/engine_darwin_test.go
+++ b/privatevoice.src/internal/engine/engine_darwin_test.go
@@ -49,3 +49,50 @@
t.Fatalf("model type = %q, want %q", config.ModelConfig.ModelType, model.BackendNemoTransducer)
}
}
+
+func TestOfflineConfigForQwen3ASRUsesQwen3Config(t *testing.T) {
+ root := t.TempDir()
+ files := map[string]string{
+ "conv_frontend": filepath.Join(root, "conv_frontend.onnx"),
+ "encoder": filepath.Join(root, "encoder.int8.onnx"),
+ "decoder": filepath.Join(root, "decoder.int8.onnx"),
+ "tokenizer": filepath.Join(root, "tokenizer"),
+ }
+ resolved := model.ResolvedModel{
+ ModelID: model.Qwen3ASRModelID,
+ BackendKind: model.BackendQwen3ASR,
+ Profile: model.ModelProfile{
+ DisplayName: "Qwen3-ASR",
+ BackendKind: model.BackendQwen3ASR,
+ ProviderOrder: []string{"cpu"},
+ NumThreads: 4,
+ },
+ Files: files,
+ }
+
+ config, err := offlineConfigForResolvedModel(resolved, "cpu")
+ if err != nil {
+ t.Fatal(err)
+ }
+ if config.ModelConfig.Qwen3ASR.ConvFrontend != files["conv_frontend"] {
+ t.Fatalf("conv frontend = %q, want %q", config.ModelConfig.Qwen3ASR.ConvFrontend, files["conv_frontend"])
+ }
+ if config.ModelConfig.Qwen3ASR.Encoder != files["encoder"] {
+ t.Fatalf("encoder = %q, want %q", config.ModelConfig.Qwen3ASR.Encoder, files["encoder"])
+ }
+ if config.ModelConfig.Qwen3ASR.Decoder != files["decoder"] {
+ t.Fatalf("decoder = %q, want %q", config.ModelConfig.Qwen3ASR.Decoder, files["decoder"])
+ }
+ if config.ModelConfig.Qwen3ASR.Tokenizer != files["tokenizer"] {
+ t.Fatalf("tokenizer = %q, want %q", config.ModelConfig.Qwen3ASR.Tokenizer, files["tokenizer"])
+ }
+ if config.ModelConfig.Qwen3ASR.MaxTotalLen != 1024 {
+ t.Fatalf("max total len = %d, want 1024", config.ModelConfig.Qwen3ASR.MaxTotalLen)
+ }
+ if config.ModelConfig.Qwen3ASR.MaxNewTokens != 256 {
+ t.Fatalf("max new tokens = %d, want 256", config.ModelConfig.Qwen3ASR.MaxNewTokens)
+ }
+ if config.ModelConfig.Tokens != "" {
+ t.Fatalf("tokens = %q, want empty for qwen3-asr", config.ModelConfig.Tokens)
+ }
+}
--
Gitblit v1.9.3