10 references to SpecialTokens
Microsoft.ML.Tokenizers.Tests (10)
LlamaTests.cs (5)
433normalizer = new SentencePieceNormalizer(removeExtraWhiteSpaces: false, addDummyPrefix: true, escapeWhiteSpaces: true, treatWhitespaceAsSuffix: false, specialTokens: (_llamaPhi3Tokenizer as LlamaTokenizer)!.SpecialTokens); 441normalizer = new SentencePieceNormalizer(removeExtraWhiteSpaces: false, addDummyPrefix: true, escapeWhiteSpaces: true, treatWhitespaceAsSuffix: true, specialTokens: (_llamaPhi3Tokenizer as LlamaTokenizer)!.SpecialTokens); 654Assert.True(tokenizer.SpecialTokens is not null); 662foreach (var kvp in tokenizer.SpecialTokens) 684string[] specialTokens = tokenizer.SpecialTokens.Keys.ToArray();
UnigramTests.cs (5)
459expectedIds[ids.Length + 1] = _unigramTokenizerWithSpecialTokens.SpecialTokens!["<pad>"]; 479expectedIds[shiftedIds.Length + 1] = _unigramTokenizerFromJson.SpecialTokens!["<pad>"]; 668expectedIds[ids.Length + 1] = _unigramTokenizerWithSpecialTokens.SpecialTokens!["<pad>"]; 687expectedIds[shiftedIds.Length + 1] = _unigramTokenizerFromJson.SpecialTokens!["<pad>"]; 830Assert.Equal(specialTokens, _unigramTokenizerFromJson.SpecialTokens);