26 lines
1.2 KiB
PowerShell
26 lines
1.2 KiB
PowerShell
# Stages the Parakeet TDT 0.6B v2 int8 bundle (sherpa-onnx export) for the
|
|
# asr_bench harness (test_parakeet branch). Bench-only artifact: lives next to
|
|
# the app's models but under bench-parakeet\, never read by the app itself.
|
|
$ErrorActionPreference = "Stop"
|
|
$dest = "$env:LOCALAPPDATA\WhispAssist\models\bench-parakeet"
|
|
$url = "https://github.com/k2-fsa/sherpa-onnx/releases/download/asr-models/sherpa-onnx-nemo-parakeet-tdt-0.6b-v2-int8.tar.bz2"
|
|
$cache = "$env:LOCALAPPDATA\WhispAssist\models\bench-cache"
|
|
New-Item -ItemType Directory -Force $cache | Out-Null
|
|
$archive = Join-Path $cache "parakeet-tdt-0.6b-v2-int8.tar.bz2"
|
|
|
|
if (Test-Path (Join-Path $dest "tokens.txt")) {
|
|
Write-Output "already staged: $dest"
|
|
exit 0
|
|
}
|
|
if (-not (Test-Path $archive)) {
|
|
Write-Output "downloading $url"
|
|
curl.exe -sL -o "$archive.part" $url
|
|
if ($LASTEXITCODE -ne 0) { throw "download failed ($LASTEXITCODE)" }
|
|
Move-Item -Force "$archive.part" $archive
|
|
}
|
|
New-Item -ItemType Directory -Force $dest | Out-Null
|
|
tar -xjf $archive -C $dest --strip-components=1
|
|
if ($LASTEXITCODE -ne 0) { throw "extract failed ($LASTEXITCODE)" }
|
|
Get-ChildItem $dest | Select-Object Name, Length
|
|
Write-Output "staged: $dest"
|