Skip to content

Commit db672e1

Browse files
authored
Merge pull request #461 from Damian89/patch-1
Update README.md (Fix for #460)
2 parents 41dc95e + be156af commit db672e1

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/reranker/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ from LM_Cocktail import mix_models, mix_models_with_data
7272
# Mix fine-tuned model and base model; then save it to output_path: ./mixed_model_1
7373
model = mix_models(
7474
model_names_or_paths=["BAAI/bge-reranker-base", "your_fine-tuned_model"],
75-
model_type='encoder',
75+
model_type='reranker',
7676
weights=[0.5, 0.5], # you can change the weights to get a better trade-off.
7777
output_path='./mixed_model_1')
7878
```

0 commit comments

Comments
 (0)