Skip to content

Commit 6352265

Browse files
committed
Fix T5Tokenizer regression
1 parent a25abce commit 6352265

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

TensorStack.TextGeneration/TensorStack.TextGeneration.csproj

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
<!--Common Packages-->
1919
<ItemGroup>
2020
<PackageReference Include="MathNet.Numerics" Version="5.0.0" />
21-
<PackageReference Include="Microsoft.ML.Tokenizers" Version="2.0.0" />
21+
<PackageReference Include="Microsoft.ML.Tokenizers" Version="2.0.0-preview.1.25125.4" />
2222
<PackageReference Include="System.Linq.Async" Version="6.0.3" />
2323
</ItemGroup>
2424

TensorStack.TextGeneration/Tokenizers/T5Tokenizer.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ private SentencePieceTokenizer CreateTokenizer()
143143
var specialTokens = GetSpecialTokens(_configuration.Path);
144144
using (var fileStream = File.OpenRead(_configuration.Path))
145145
{
146-
return SentencePieceTokenizer.Create(fileStream, addBeginningOfSentence: false, addEndOfSentence: true, specialTokens);
146+
return SentencePieceTokenizer.Create(fileStream, addBeginOfSentence: false, addEndOfSentence: true, specialTokens);
147147
}
148148
}
149149

0 commit comments

Comments
 (0)