File tree Expand file tree Collapse file tree
src/test/java/net/ladenthin/llama Expand file tree Collapse file tree Original file line number Diff line number Diff line change 77
88import static org .junit .jupiter .api .Assertions .*;
99
10+ import java .io .File ;
1011import java .util .List ;
1112import java .util .Map ;
1213import net .ladenthin .llama .parameters .ModelParameters ;
1314import net .ladenthin .llama .value .LlamaOutput ;
1415import net .ladenthin .llama .value .Pair ;
1516import org .junit .jupiter .api .AfterAll ;
17+ import org .junit .jupiter .api .Assumptions ;
1618import org .junit .jupiter .api .BeforeAll ;
1719import org .junit .jupiter .api .Test ;
1820
@@ -30,6 +32,9 @@ public class RerankingModelTest {
3032
3133 @ BeforeAll
3234 public static void setup () {
35+ Assumptions .assumeTrue (
36+ new File ("models/jina-reranker-v1-tiny-en-Q4_0.gguf" ).exists (),
37+ "Reranking model not available, skipping tests" );
3338 int gpuLayers = Integer .getInteger (TestConstants .PROP_TEST_NGL , TestConstants .DEFAULT_TEST_NGL );
3439 model = new LlamaModel (new ModelParameters ()
3540 .setCtxSize (128 )
You can’t perform that action at this time.
0 commit comments