File tree Expand file tree Collapse file tree 3 files changed +19
-5
lines changed Expand file tree Collapse file tree 3 files changed +19
-5
lines changed Original file line number Diff line number Diff line change 5656 </ItemGroup >
5757
5858 <PropertyGroup >
59- <BinaryReleaseId >c35e586ea5722184 </BinaryReleaseId >
59+ <BinaryReleaseId >958367bf530d943a90 </BinaryReleaseId >
6060 </PropertyGroup >
6161
6262 <PropertyGroup >
Original file line number Diff line number Diff line change @@ -768,17 +768,32 @@ internal ModelTokens(SafeLlamaModelHandle model)
768768 /// <summary>
769769 /// Codellama beginning of infill prefix
770770 /// </summary>
771- public LLamaToken ? InfillPrefix => Normalize ( llama_token_prefix ( _model ) ) ;
771+ public LLamaToken ? InfillPrefix => Normalize ( llama_token_fim_pre ( _model ) ) ;
772772
773773 /// <summary>
774774 /// Codellama beginning of infill middle
775775 /// </summary>
776- public LLamaToken ? InfillMiddle => Normalize ( llama_token_middle ( _model ) ) ;
776+ public LLamaToken ? InfillMiddle => Normalize ( llama_token_fim_mid ( _model ) ) ;
777777
778778 /// <summary>
779779 /// Codellama beginning of infill suffix
780780 /// </summary>
781- public LLamaToken ? InfillSuffix => Normalize ( llama_token_suffix ( _model ) ) ;
781+ public LLamaToken ? InfillSuffix => Normalize ( llama_token_fim_suf ( _model ) ) ;
782+
783+ /// <summary>
784+ /// Codellama pad
785+ /// </summary>
786+ public LLamaToken ? InfillPad => Normalize ( llama_token_fim_pad ( _model ) ) ;
787+
788+ /// <summary>
789+ /// Codellama rep
790+ /// </summary>
791+ public LLamaToken ? InfillRep => Normalize ( llama_token_fim_rep ( _model ) ) ;
792+
793+ /// <summary>
794+ /// Codellama rep
795+ /// </summary>
796+ public LLamaToken ? InfillSep => Normalize ( llama_token_fim_sep ( _model ) ) ;
782797
783798 /// <summary>
784799 /// Codellama end of infill middle
Original file line number Diff line number Diff line change @@ -141,7 +141,6 @@ protected override SafeLLamaSamplerChainHandle CreateChain(SafeLLamaContextHandl
141141 chain . AddMinP ( MinP , MinKeep ) ;
142142 chain . AddTemperature ( Temperature ) ;
143143
144- chain . AddSoftmax ( ) ;
145144 chain . AddDistributionSampler ( Seed ) ;
146145
147146 return chain ;
You can’t perform that action at this time.
0 commit comments