mirror of
https://github.com/ggerganov/whisper.cpp.git
synced 2024-12-24 06:46:37 +00:00
3a27b2b91b
* Remove Whisper::Model.[] * Fix Whisper::Model::URI#request * Make Whisper::Context#initialize accept pre-converted model name * Use downloading pre-converted model feature for testing * Update README * Remove unnecessary task * Move whisper/model.rb -> whisper/model/uri.rb * Update document comment of Whisper::Context#initialize * Don't show download progress when not tty * Pass String to raise * Use cache model file if download fails * Add test for auto download * Specify required Ruby version * Fix a typo * Remove unnecessary flags * Initialize Whisper::Params#diarize explicitely * Remove redundant code from README for simplicity * Add Whisper::Params#no_speech_thold attribute * Add test for Whisper::Params#no_speech_thold
65 lines
1.5 KiB
Ruby
65 lines
1.5 KiB
Ruby
require 'rake/clean'
|
|
require "bundler/gem_tasks"
|
|
require "rake/testtask"
|
|
require_relative "extsources"
|
|
|
|
SOURCES = FileList[]
|
|
|
|
EXTSOURCES.each do |src|
|
|
basename = src.pathmap("%f")
|
|
dest = basename == "LICENSE" ? basename : src.pathmap("%{../..,ext}p")
|
|
dir = dest.pathmap("%d")
|
|
file src
|
|
directory dir
|
|
file dest => [src, dir] do |t|
|
|
cp t.source, t.name
|
|
end
|
|
SOURCES.include dest
|
|
end
|
|
|
|
CLEAN.include SOURCES
|
|
CLEAN.include FileList["ext/*.o", "ext/*.metal", "ext/whisper.{so,bundle,dll}"]
|
|
|
|
task build: ["ext/Makefile", "ext/ruby_whisper.h", "ext/ruby_whisper.cpp", "whispercpp.gemspec"]
|
|
|
|
directory "pkg"
|
|
CLOBBER.include "pkg"
|
|
|
|
LIB_NAME = "whisper".ext(RbConfig::CONFIG["DLEXT"])
|
|
SO_FILE = File.join("ext", LIB_NAME)
|
|
LIB_FILE = File.join("lib", LIB_NAME)
|
|
|
|
file "ext/Makefile" => ["ext/extconf.rb", "ext/ruby_whisper.h", "ext/ruby_whisper.cpp"] + SOURCES do |t|
|
|
Dir.chdir "ext" do
|
|
ruby "extconf.rb"
|
|
end
|
|
end
|
|
|
|
file SO_FILE => "ext/Makefile" do |t|
|
|
Dir.chdir "ext" do
|
|
sh "make"
|
|
end
|
|
end
|
|
CLEAN.include SO_FILE
|
|
|
|
directory "lib"
|
|
file LIB_FILE => [SO_FILE, "lib"] do |t|
|
|
copy t.source, t.name
|
|
end
|
|
CLEAN.include LIB_FILE
|
|
|
|
Rake::TestTask.new do |t|
|
|
t.test_files = FileList["tests/test_*.rb"]
|
|
end
|
|
|
|
TEST_MEMORY_VIEW = "tests/jfk_reader/jfk_reader.#{RbConfig::CONFIG['DLEXT']}"
|
|
file TEST_MEMORY_VIEW => "tests/jfk_reader/jfk_reader.c" do |t|
|
|
Dir.chdir "tests/jfk_reader" do
|
|
ruby "extconf.rb"
|
|
sh "make"
|
|
end
|
|
end
|
|
CLEAN.include "tests/jfk_reader/jfk_reader.{o,#{RbConfig::CONFIG['DLEXT']}}"
|
|
|
|
task test: [LIB_FILE, TEST_MEMORY_VIEW]
|