Add Parakeet model shortcuts
This commit is contained in:
parent
ca3a565fef
commit
5b2bdc84fc
|
|
@ -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|
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue