Register parakeet-tdt-0.6b-v3

This commit is contained in:
Kitaiti Makoto 2026-05-22 23:45:01 +09:00
parent 9ed7a04394
commit 93af6b2080
1 changed files with 7 additions and 1 deletions

View File

@ -216,8 +216,14 @@ module Whisper
@pre_converted_models[name] = URI.new("https://huggingface.co/ggml-org/whisper-vad/resolve/main/ggml-#{name}.bin")
end
%w[
parakeet-tdt-0.6b-v3
].each do |name|
@pre_converted_models[name] = URI.new("https://huggingface.co/danbev/parakeet/resolve/main/#{name}.bin")
end
@coreml_compiled_models = @pre_converted_models.each_with_object({}) {|(name, uri), models|
next if name.end_with?("-tdrz") || name.start_with?("silero-")
next if name.end_with?("-tdrz") || name.start_with?("silero-") || name.start_with?("parakeet-")
if matched = name.match(/\A(?<name>.*)-q\d_\d\z/)
name = matched[:name]