//go:build darwin && !monterey_compat
|
|
package engine
|
|
import (
|
"path/filepath"
|
"testing"
|
|
"voicesnap/internal/model"
|
)
|
|
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)
|
}
|
}
|
