1 instantiation of ModelProto
Microsoft.ML.Tokenizers (1)
SentencepieceModel.cs (1)
180ModelProto result = new();
44 references to ModelProto
Microsoft.ML.Tokenizers (44)
Model\LlamaTokenizer.cs (3)
19internal LlamaTokenizer(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? addedTokens = null) : base(modelProto, addBos, addEos, addedTokens) 40ModelProto modelProto = ModelProto.Parser.ParseFrom(modelStream);
Model\SentencePieceBaseModel.cs (1)
18internal SentencePieceBaseModel(ModelProto modelProto, bool addBos = false, bool addEos = false, IReadOnlyDictionary<string, int>? specialTokens = null)
Model\SentencePieceBpeModel.cs (11)
26internal SentencePieceBpeModel(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? specialTokens = null) : base(modelProto, addBos, addEos, specialTokens) 34if (piece.Type == ModelProto.Types.SentencePiece.Types.Type.Byte) 180if (type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused) 268if (id.Type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused || 458if (type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused) 573if (id.Type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused || 726if (type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused) 832if (id.Type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused || 1002if (type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused) 1108if (id.Type != (byte)ModelProto.Types.SentencePiece.Types.Type.Unused || 1208if (leftId.Type == (byte)ModelProto.Types.SentencePiece.Types.Type.Unused)
Model\SentencePieceTokenizer.cs (3)
24internal SentencePieceTokenizer(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? specialTokens = null) 457ModelProto modelProto = ModelProto.Parser.ParseFrom(modelStream);
Model\SentencePieceUnigramModel.cs (21)
22private readonly (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[] _vocabReverse; 30public SentencePieceUnigramModel(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? specialTokens = null) : base(modelProto, addBos, addEos, specialTokens) 41_vocabReverse = new (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[modelProto.Pieces.Count]; 48if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Normal || 49modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.UserDefined || 50modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unused) 59else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Byte) 63else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unknown) 66_vocabReverse[i] = (modelProto.Pieces[i].Piece, modelProto.Pieces[i].Score, ModelProto.Types.SentencePiece.Types.Type.Unknown); 88_vocabReverse[modelProto.TrainerSpec.BosId] = (modelProto.TrainerSpec.BosPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 89_vocabReverse[modelProto.TrainerSpec.EosId] = (modelProto.TrainerSpec.EosPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 90_vocabReverse[modelProto.TrainerSpec.UnkId] = (modelProto.TrainerSpec.UnkPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Unknown); 95_vocabReverse[modelProto.TrainerSpec.PadId] = (modelProto.TrainerSpec.PadPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 174_vocabReverse = new (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[pieces!.Count]; 197_vocabReverse[i] = (piece, score, ModelProto.Types.SentencePiece.Types.Type.Unknown); 201_vocabReverse[i] = (piece, score, ModelProto.Types.SentencePiece.Types.Type.Control); 205_vocabReverse[i] = (piece, score, ModelProto.Types.SentencePiece.Types.Type.Normal); 239_vocabReverse[unkId] = (unkToken, 0f, ModelProto.Types.SentencePiece.Types.Type.Unknown); 253_vocabReverse[controlId] = (piece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 781if (_vocabReverse[ret].Type == ModelProto.Types.SentencePiece.Types.Type.Unused) 791float score = _vocabReverse[ret].Type == ModelProto.Types.SentencePiece.Types.Type.UserDefined ? length * _maxScore - 0.1f : _vocabReverse[ret].Score;
SentencepieceModel.cs (5)
178internal static ModelProto Parse(byte[] data, int start, int end) 180ModelProto result = new(); 288internal ModelProto ParseFrom(Stream stream) 302return ModelProto.Parse(segment.Array!, start, end); 318return ModelProto.Parse(ms.GetBuffer(), 0, (int)ms.Length);