Skip to content

Commit 424a736

Browse files
authored
Fixed some failing tests, it looks like there's a min context size which is why these were failing now. (#1)
1 parent 48f109a commit 424a736

2 files changed

Lines changed: 4 additions & 4 deletions

File tree

LLama.Unittest/LLamaContextTests.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ public LLamaContextTests()
1313
{
1414
var @params = new ModelParams(Constants.GenerativeModelPath2)
1515
{
16-
ContextSize = 128,
16+
ContextSize = 512,
1717
BatchSize = 8,
1818
UBatchSize = 8,
1919
SeqMax = 1,
@@ -33,7 +33,7 @@ public void Dispose()
3333
[Fact]
3434
public void CheckProperties()
3535
{
36-
Assert.Equal(128u, _context.ContextSize);
36+
Assert.Equal(512u, _context.ContextSize);
3737
Assert.Equal(960, _context.EmbeddingSize);
3838
Assert.Equal(49152, _context.Vocab.Count);
3939
}

LLama.Unittest/LLamaContextWithCustomLoggerTests.cs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ public LLamaContextWithCustomLoggerTests()
3030
{
3131
var @params = new ModelParams(Constants.GenerativeModelPath2)
3232
{
33-
ContextSize = 128,
33+
ContextSize = 512,
3434
GpuLayerCount = Constants.CIGpuLayerCount,
3535
};
3636

@@ -55,7 +55,7 @@ public void Dispose()
5555
[Fact]
5656
public void CheckProperties()
5757
{
58-
Assert.Equal(128u, _context.ContextSize);
58+
Assert.Equal(512u, _context.ContextSize);
5959
Assert.Equal(960, _context.EmbeddingSize);
6060
Assert.Equal(49152, _context.Vocab.Count);
6161
}

0 commit comments

Comments
 (0)