Skip to content

Commit 01a6f83

Browse files
author
Vaijanath Rao
committed
updating the workflow and reranking
1 parent e9c3de7 commit 01a6f83

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

.github/workflows/ci.yml

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@ jobs:
2727
run: curl -L ${MODEL_URL} --create-dirs -o models/${MODEL_NAME}
2828
- name: Download reranking model
2929
run: curl -L ${RERANKING_MODEL_URL} --create-dirs -o models/${RERANKING_MODEL_NAME}
30+
- name: List files in models directory
31+
run: ls -l models/
3032
- name: Run tests
3133
run: mvn test
3234
- if: failure()
@@ -61,7 +63,8 @@ jobs:
6163
run: curl -L ${MODEL_URL} --create-dirs -o models/${MODEL_NAME}
6264
- name: Download reranking model
6365
run: curl -L ${RERANKING_MODEL_URL} --create-dirs -o models/${RERANKING_MODEL_NAME}
64-
66+
- name: List files in models directory
67+
run: ls -l models/
6568
- name: Run tests
6669
run: mvn test
6770
- if: failure()
@@ -88,7 +91,8 @@ jobs:
8891
run: curl -L $env:MODEL_URL --create-dirs -o models/$env:MODEL_NAME
8992
- name: Download reranking model
9093
run: curl -L ${RERANKING_MODEL_URL} --create-dirs -o models/${RERANKING_MODEL_NAME}
91-
94+
- name: List files in models directory
95+
run: ls -l models/
9296
- name: Run tests
9397
run: mvn test
9498
- if: failure()

src/test/java/de/kherud/llama/RerankingModelTest.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ public class RerankingModelTest {
1818
@BeforeClass
1919
public static void setup() {
2020
model = new LlamaModel(
21-
new ModelParameters().setCtxSize(128).setModel("models/jina-reranker-v1-tiny-en.Q4_K_M.gguf")
21+
new ModelParameters().setCtxSize(128).setModel("models/jina-reranker-v1-tiny-en-Q4_0.gguf")
2222
.setGpuLayers(43).enableReranking().enableLogTimestamps().enableLogPrefix());
2323
}
2424

0 commit comments

Comments
 (0)