1 instantiation of ModelProto
Microsoft.ML.Tokenizers (1)
SentencepieceModel.cs (1)
180ModelProto result = new();
38 references to ModelProto
Microsoft.ML.Tokenizers (38)
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)
23internal SentencePieceTokenizer(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? specialTokens = null) 451ModelProto modelProto = ModelProto.Parser.ParseFrom(modelStream);
Model\SentencePieceUnigramModel.cs (15)
21private readonly (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[] _vocabReverse; 27public SentencePieceUnigramModel(ModelProto modelProto, bool addBos, bool addEos, IReadOnlyDictionary<string, int>? specialTokens = null) : base(modelProto, addBos, addEos, specialTokens) 38_vocabReverse = new (string Piece, float Score, ModelProto.Types.SentencePiece.Types.Type Type)[modelProto.Pieces.Count]; 45if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Normal || 46modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.UserDefined || 47modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unused) 56else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Byte) 60else if (modelProto.Pieces[i].Type == ModelProto.Types.SentencePiece.Types.Type.Unknown) 63_vocabReverse[i] = (modelProto.Pieces[i].Piece, modelProto.Pieces[i].Score, ModelProto.Types.SentencePiece.Types.Type.Unknown); 85_vocabReverse[modelProto.TrainerSpec.BosId] = (modelProto.TrainerSpec.BosPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 86_vocabReverse[modelProto.TrainerSpec.EosId] = (modelProto.TrainerSpec.EosPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 87_vocabReverse[modelProto.TrainerSpec.UnkId] = (modelProto.TrainerSpec.UnkPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Unknown); 92_vocabReverse[modelProto.TrainerSpec.PadId] = (modelProto.TrainerSpec.PadPiece, 0f, ModelProto.Types.SentencePiece.Types.Type.Control); 506if (_vocabReverse[ret].Type == ModelProto.Types.SentencePiece.Types.Type.Unused) 516float 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);