Skip to content

Commit 2bc394e

Browse files
committed
fixed to work with new binaries
1 parent fb168ec commit 2bc394e

File tree

3 files changed

+19
-5
lines changed

3 files changed

+19
-5
lines changed

LLama/LLamaSharp.csproj

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@
5656
</ItemGroup>
5757

5858
<PropertyGroup>
59-
<BinaryReleaseId>c35e586ea5722184</BinaryReleaseId>
59+
<BinaryReleaseId>958367bf530d943a90</BinaryReleaseId>
6060
</PropertyGroup>
6161

6262
<PropertyGroup>

LLama/Native/SafeLlamaModelHandle.cs

Lines changed: 18 additions & 3 deletions
Original file line numberDiff line numberDiff 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

LLama/Sampling/DefaultSamplingPipeline.cs

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff 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;

0 commit comments

Comments
 (0)