Skip to content

Commit

Permalink
Merge branch 'feat/diarization-v1' of https://github.com/thewh1teagle…
Browse files Browse the repository at this point in the history
…/vibe into feat/diarization-v1
  • Loading branch information
thewh1teagle committed Aug 8, 2024
2 parents f5cff91 + 4c291a1 commit 4532653
Show file tree
Hide file tree
Showing 3 changed files with 16 additions and 5 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@
- ⏳ Transcribe system audio
- 🎤 Transcribe from microphone
- 🖥️ CLI support: Use Vibe directly from the command line interface! (see `--help`)
- 👥 ~Speaker diarization~ (coming soon)
- 👥 Speaker diarization
- 📱 ~iOS & Android support~ (coming soon)
- 📥 Integrate custom models from your own site: Use `vibe://download/?url=<model url>`
- 📹 Choose caption length optimized for videos / reels
Expand Down
2 changes: 1 addition & 1 deletion desktop/src-tauri/tauri.conf.json
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
{
"$schema": "./gen/schemas/desktop-schema.json",
"productName": "vibe",
"version": "2.3.0",
"version": "2.4.0-beta.0",
"identifier": "github.com.thewh1teagle.vibe",
"app": {
"windows": [],
Expand Down
17 changes: 14 additions & 3 deletions scripts/pre_build.js
Original file line number Diff line number Diff line change
Expand Up @@ -205,9 +205,14 @@ if (hasFeature('rocm')) {
}

// Diarization
if (!fs.exists(config.diarization.embedModelFilename)) {
await $`wget -nc --show-progress ${config.diarization.embedModelUrl} -O ${config.diarization.embedModelFilename}`
await $`wget -nc --show-progress ${config.diarization.segmentModelUrl} -O ${config.diarization.segmentModelFilename}`
if (!(await fs.exists(config.diarization.embedModelFilename))) {
if (platform == 'windows') {
await $`C:\\msys64\\usr\\bin\\wget.exe -nc --show-progress ${config.diarization.embedModelUrl} -O ${config.diarization.embedModelFilename}`
await $`C:\\msys64\\usr\\bin\\wget.exe -nc --show-progress ${config.diarization.segmentModelUrl} -O ${config.diarization.segmentModelFilename}`
} else {
await $`wget -nc --show-progress ${config.diarization.embedModelUrl} -O ${config.diarization.embedModelFilename}`
await $`wget -nc --show-progress ${config.diarization.segmentModelUrl} -O ${config.diarization.segmentModelFilename}`
}
}

// Development hints
Expand All @@ -220,6 +225,7 @@ if (!process.env.GITHUB_ENV) {
}
console.log('bun install')
if (platform == 'windows') {
console.log(`$env:RUSTFLAGS = "-C target-feature=+crt-static"`)
console.log(`$env:FFMPEG_DIR = "${exports.ffmpeg}"`)
console.log(`$env:OPENBLAS_PATH = "${exports.openBlas}"`)
console.log(`$env:LIBCLANG_PATH = "${exports.libClang}"`)
Expand Down Expand Up @@ -269,6 +275,11 @@ if (process.env.GITHUB_ENV) {
await fs.appendFile(process.env.GITHUB_ENV, embed_metal)
}
if (platform == 'windows') {
// ort + whisper.cpp + fbank-rs... something there requires static linking of msvc
const rustFlags = `RUSTFLAGS=-C target-feature=+crt-static\n`
console.log('Adding ENV', rustFlags)
await fs.appendFile(process.env.GITHUB_ENV, rustFlags)

const openblas = `OPENBLAS_PATH=${exports.openBlas}\n`
console.log('Adding ENV', openblas)
await fs.appendFile(process.env.GITHUB_ENV, openblas)
Expand Down

0 comments on commit 4532653

Please sign in to comment.