Mirrors/SubtitleEdit
1
0
Fork 0
mirror of https://github.com/SubtitleEdit/subtitleedit.git synced 2024-11-21 18:52:36 +01:00
Code Issues Projects Releases Wiki Activity

Update Whisper Norwegian models - thx Jostein :)

Browse Source
This commit is contained in:
Nikolaj Olsson 2024-04-19 23:29:46 +02:00
parent 264da9188d
commit 57efc1bd18
3 changed files with 70 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/libse/AudioToText/WhisperCppModel.cs
View File

@ -40,7 +40,6 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
}
private const string DownloadUrlPrefix = "https://huggingface.co/ggerganov/whisper.cpp/resolve/main/";
private const string DownloadUrlNbPrefix = "https://huggingface.co/NbAiLab/nb-whisper-large-beta/resolve/main/";
public WhisperModel[] Models => new[]
{
@ -164,11 +163,41 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
Size = "2.88 GB",
Urls = new []{ DownloadUrlPrefix + "ggml-large-v3-q5_0.bin" },
},
new WhisperModel
{
Name = "ggml-model",
Size = "3.1 GB Norwegian NB, beta",
Urls = new []{ DownloadUrlNbPrefix + "ggml-model.bin" },
Name = "tiny.nb",
Rename = true,
Size = "78 MB Norwegian",
Urls = new []{ "https://huggingface.co/NbAiLab/nb-whisper-tiny/resolve/main/ggml-model.bin" },
},
new WhisperModel
{
Name = "base.nb",
Rename = true,
Size = "148 MB Norwegian",
Urls = new []{ "https://huggingface.co/NbAiLab/nb-whisper-base/resolve/main/ggml-model.bin" },
},
new WhisperModel
{
Name = "small.nb",
Rename = true,
Size = "488 MB Norwegian",
Urls = new []{ "https://huggingface.co/NbAiLab/nb-whisper-small/resolve/main/ggml-model.bin" },
},
new WhisperModel
{
Name = "medium.nb",
Rename = true,
Size = "1.5 GB Norwegian",
Urls = new []{ "https://huggingface.co/NbAiLab/nb-whisper-medium/resolve/main/ggml-model.bin" },
},
new WhisperModel
{
Name = "large.nb",
Rename = true,
Size = "3.1 GB Norwegian",
Urls = new []{ "https://huggingface.co/NbAiLab/nb-whisper-large/resolve/main/ggml-model.bin" },
},
};
}

1
src/libse/AudioToText/WhisperModel.cs
View File

@ -8,6 +8,7 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
public string[] Urls { get; set; }
public string Size { get; set; }
public string Name { get; set; }
public bool Rename { get; set; }
public string Folder { get; set; }
public bool AlreadyDownloaded { get; set; }
public long Bytes { get; set; }

36
src/libse/AudioToText/WhisperPurfviewFasterWhisperModel.cs
View File

@ -146,6 +146,42 @@ namespace Nikse.SubtitleEdit.Core.AudioToText
Urls = MakeUrls("https://huggingface.co/Systran/faster-distil-whisper-large-v3/resolve/main"),
Folder = "faster-distil-whisper-large-v3",
},
new WhisperModel
{
Name = "tiny.nb",
Size = "151 MB Norwegian",
Urls = MakeUrls("https://huggingface.co/NbAiLab/nb-whisper-tiny/resolve/main"),
Folder = "faster-whisper-tiny.nb",
},
new WhisperModel
{
Name = "base.nb",
Size = "290 MB Norwegian",
Urls = MakeUrls("https://huggingface.co/NbAiLab/nb-whisper-base/resolve/main"),
Folder = "faster-whisper-base.nb",
},
new WhisperModel
{
Name = "small.nb",
Size = "151 MB Norwegian",
Urls = MakeUrls("https://huggingface.co/NbAiLab/nb-whisper-small/resolve/main"),
Folder = "faster-whisper-small.nb",
},
new WhisperModel
{
Name = "medium.nb",
Size = "3.1 GB Norwegian",
Urls = MakeUrls("https://huggingface.co/NbAiLab/nb-whisper-medium/resolve/main"),
Folder = "faster-whisper-medium.nb",
},
new WhisperModel
{
Name = "large.nb",
Size = "6.2 GB Norwegian",
Urls = MakeUrls("https://huggingface.co/NbAiLab/nb-whisper-large/resolve/main"),
Folder = "faster-whisper-large.nb",
},
};
private string[] MakeUrls(string baseUrl)