File tree Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Expand file tree Collapse file tree 2 files changed +12
-2
lines changed Original file line number Diff line number Diff line change 7
7
8
8
namespace LLama . Unittest ;
9
9
10
- public sealed class LLamaRerankerTests
10
+ public sealed class LLamaRerankerTests : IDisposable
11
11
{
12
12
private readonly ITestOutputHelper _testOutputHelper ;
13
13
private readonly LLamaReranker _reranker ;
@@ -26,6 +26,11 @@ public LLamaRerankerTests(ITestOutputHelper testOutputHelper)
26
26
_reranker = new LLamaReranker ( weights , @params ) ;
27
27
}
28
28
29
+ public void Dispose ( )
30
+ {
31
+ _reranker . Dispose ( ) ;
32
+ }
33
+
29
34
[ Fact ]
30
35
public async Task CompareRerankingScore ( )
31
36
{
Original file line number Diff line number Diff line change 7
7
8
8
namespace LLama . Unittest . Native ;
9
9
10
- public class SafeLlamaModelHandleVocabularyTests
10
+ public class SafeLlamaModelHandleVocabularyTests : IDisposable
11
11
{
12
12
private readonly LLamaWeights _model ;
13
13
@@ -22,6 +22,11 @@ public SafeLlamaModelHandleVocabularyTests()
22
22
_model = LLamaWeights . LoadFromFile ( @params ) ;
23
23
}
24
24
25
+ public void Dispose ( )
26
+ {
27
+ _model . Dispose ( ) ;
28
+ }
29
+
25
30
[ Fact ]
26
31
public void GetLLamaTokenString ( )
27
32
{
You can’t perform that action at this time.
0 commit comments