diff --git a/bindings/ruby/lib/whisper/model/uri.rb b/bindings/ruby/lib/whisper/model/uri.rb index 5cd24ddc5..fca750278 100644 --- a/bindings/ruby/lib/whisper/model/uri.rb +++ b/bindings/ruby/lib/whisper/model/uri.rb @@ -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| diff --git a/bindings/ruby/test/test_parakeet_callback.rb b/bindings/ruby/test/test_parakeet_callback.rb index ed04befec..1209e960f 100644 --- a/bindings/ruby/test/test_parakeet_callback.rb +++ b/bindings/ruby/test/test_parakeet_callback.rb @@ -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 diff --git a/bindings/ruby/test/test_parakeet_context.rb b/bindings/ruby/test/test_parakeet_context.rb index 52d75dce2..2d039ce75 100644 --- a/bindings/ruby/test/test_parakeet_context.rb +++ b/bindings/ruby/test/test_parakeet_context.rb @@ -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 diff --git a/bindings/ruby/test/test_parakeet_segment.rb b/bindings/ruby/test/test_parakeet_segment.rb index 37b5806d9..d5b99bd5e 100644 --- a/bindings/ruby/test/test_parakeet_segment.rb +++ b/bindings/ruby/test/test_parakeet_segment.rb @@ -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 diff --git a/bindings/ruby/test/test_parakeet_token.rb b/bindings/ruby/test/test_parakeet_token.rb index db4751690..6f0b8b5a3 100644 --- a/bindings/ruby/test/test_parakeet_token.rb +++ b/bindings/ruby/test/test_parakeet_token.rb @@ -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