Skip to content

Commit 49edaf6

Browse files
committed
Add support for models with Byte Pair Encoding (BPE) vocabulary type
1 parent b8bb96b commit 49edaf6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

quantize_weights_for_llama.cpp.ps1

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,11 @@ ForEach ($repositoryName in $repositoryDirectories) {
5555

5656
$convertParameters = "--outfile `"${unquantizedModelPath}`" `"${sourceDirectoryPath}`""
5757

58+
# Some models have a Byte Pair Encoding (BPE) vocabulary type.
59+
if (@("Smaug-72B-v0.1").Contains($repositoryName)) {
60+
$convertParameters = "--vocab-type `"bpe`" --pad-vocab $convertParameters"
61+
}
62+
5863
Invoke-Expression "$convertCommand $convertParameters"
5964

6065
# Some model architectures have not yet been backported into

0 commit comments

Comments
 (0)