34 lines
774 B
C#
34 lines
774 B
C#
using System.Text;
|
|
using LLama.Common;
|
|
|
|
namespace LLama.Unittest
|
|
{
|
|
public sealed class BasicTest
|
|
: IDisposable
|
|
{
|
|
private readonly ModelParams _params;
|
|
private readonly LLamaWeights _model;
|
|
|
|
public BasicTest()
|
|
{
|
|
_params = new ModelParams(Constants.ModelPath)
|
|
{
|
|
ContextSize = 2048
|
|
};
|
|
_model = LLamaWeights.LoadFromFile(_params);
|
|
}
|
|
|
|
public void Dispose()
|
|
{
|
|
_model.Dispose();
|
|
}
|
|
|
|
[Fact]
|
|
public void BasicModelProperties()
|
|
{
|
|
Assert.Equal(32000, _model.VocabCount);
|
|
Assert.Equal(4096, _model.ContextSize);
|
|
Assert.Equal(4096, _model.EmbeddingSize);
|
|
}
|
|
}
|
|
} |