Add Parakeet model shortcuts

This commit is contained in:
Kitaiti Makoto 2026-06-15 04:11:09 +09:00
parent ca3a565fef
commit 5b2bdc84fc
5 changed files with 11 additions and 7 deletions

View File

@ -217,9 +217,13 @@ module Whisper
end
%w[
parakeet-tdt-0.6b-v3
parakeet-tdt-0.6b-v3-f16
parakeet-tdt-0.6b-v3-f32
parakeet-tdt-0.6b-v3-q4_0
parakeet-tdt-0.6b-v3-q4_k
parakeet-tdt-0.6b-v3-q8_0
].each do |name|
@pre_converted_models[name] = URI.new("https://huggingface.co/danbev/parakeet/resolve/main/#{name}.bin")
@pre_converted_models[name] = URI.new("https://huggingface.co/danbev/parakeet-GGUF/resolve/main/ggml-#{name}.bin")
end
@coreml_compiled_models = @pre_converted_models.each_with_object({}) {|(name, uri), models|

View File

@ -7,7 +7,7 @@ class TestParakeetCallback < TestBase
Whisper.instance_variable_set "@whisper", nil
GC.start
@params = Parakeet::Params.new
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3")
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3-q4_0")
end
def test_new_segment_callback

View File

@ -8,7 +8,7 @@ class TestParakeetContext < TestBase
Whisper.instance_variable_set "@whisper", nil
GC.start
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3")
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3-q4_0")
@params = Parakeet::Params.new
end
@ -23,7 +23,7 @@ class TestParakeetContext < TestBase
logs = ""
Parakeet.log_set proc {|level, message| logs << message}, nil
params = Parakeet::Context::Params.new(use_gpu: false)
parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3", params)
parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3-q4_0", params)
assert_instance_of Parakeet::Context, parakeet
assert_match /use gpu\s+=\s+0/, logs
ensure

View File

@ -4,7 +4,7 @@ class TestParakeetSegment < TestBase
def setup
omit "Skip not to download large model" if ENV["CI"]
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3")
@parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3-q4_0")
@parakeet.transcribe AUDIO, Parakeet::Params.new
end

View File

@ -20,7 +20,7 @@ class TestParakeetToken < TestBase
Whisper.instance_variable_set "@whisper", nil
GC.start
parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3")
parakeet = Parakeet::Context.new("parakeet-tdt-0.6b-v3-q4_0")
params = Parakeet::Params.new
parakeet.transcribe AUDIO, params
@segment = parakeet.each_segment.first