Reorder - #143

Open

XKTZ wants to merge 60 commits into

castorini:mainfrom

Commits on Aug 13, 2024

First step update - during create_prompt, use selected_index to substitude rank_start and rank_end
XKTZ
committed
60d5e7f
Allow selected index to be variable on each step
XKTZ
committed
74118c4
Added basic _get_model_function - it is wrong now, we want to return a permutation but now it is returning a string. Will need the cleanup pipeline in next step
XKTZ
committed
36bbcdc

Commits on Aug 14, 2024

Renamed indexes to indices, [indices] is renamed to indices_batch
XKTZ
committed
6d2094a
Renamed ReorderExecutor to Reorder Policy
XKTZ
committed
cc78522

Commits on Aug 15, 2024

Moved LiT5Distill to policy
XKTZ
committed
4936c8c

Commits on Aug 18, 2024

Transition rank listwise os llm to using reorder policy
XKTZ
committed
95f065a
Added reorder policy for rank listwise os and fid score
XKTZ
committed
438c8bd
Revised bug in OS LLM, add Rank GPT, deprecated old functions
XKTZ
committed
2aa8644

Commits on Aug 25, 2024

Finish the tournament sort node
XKTZ
committed
7a19b87

Commits on Aug 26, 2024

Finish tournament sort
XKTZ
committed
c8734f0

Commits on Aug 30, 2024

Finish reorganize of parameters, move window_size to ListwiseRankLLM
XKTZ
committed
7981c06

Commits on Sep 5, 2024

Added window size back
XKTZ
committed
bbbdc31
Fix Rerankers
XKTZ
committed
bb0ad2b
Merge pull request #4 from castorini/main
XKTZ
authored
62785bc
Merge branch 'reorder' into main
XKTZ
authored
9b1972a
Merge pull request #5 from XKTZ/main
XKTZ
authored
06b9f1f
Added r parameter
XKTZ
committed
28e3a23

Commits on Sep 6, 2024

Some bug fix
XKTZ
committed
0e9edce
Reformatted
XKTZ
committed
3005455

Commits on Sep 12, 2024

Added top down
XKTZ
committed
3795308
Merge pull request #6 from castorini/main
XKTZ
authored
edd1ff9
Merge pull request #7 from XKTZ/main
XKTZ
authored
89bab7c

Commits on Sep 15, 2024

Added final some stuff
XKTZ
committed
65d3508
Updated readme
XKTZ
committed
69e3d65
Clean the README
XKTZ
committed
8f4f729

Commits on Sep 16, 2024

Changed filename
XKTZ
committed
cc22c95
Updated silence
XKTZ
committed
26beeed

Commits on Sep 20, 2024

Fixed bug when batch size > 32, which is in rank_listwise_os_llm's create_prompt
XKTZ
committed
36beed1

Commits on Sep 25, 2024

Updated for removing use_tqdm=False
XKTZ
committed
406fca7

Commits on Sep 27, 2024

Update parallel for topdown
XKTZ
committed
9e7f846
Reformat
XKTZ
committed
329d6eb

Commits on Oct 3, 2024

Added vllm_chunked_prefill
XKTZ
committed
4f561e8

Commits on Oct 19, 2024

Updated README
XKTZ
committed
8470d3d
Put back step size for maintaining backward compatibility
XKTZ
committed
9caa5fb

Commits on Nov 18, 2024

Support early stop in topdown
XKTZ
committed
d7ecc8b

Commits on Nov 21, 2024

Reformat
XKTZ
committed
891ca8f
temp .gitignore
XKTZ
committed
db3a509

Commits on Dec 15, 2024

Add parameter include padding,
XKTZ
committed
743b5af

Commits on Dec 24, 2024

Merge remote-tracking branch 'origin/main' into reorder
XKTZ
committed
cca845e
Format
XKTZ
committed
66c8742
Add more fix
XKTZ
committed
acf8012

Commits on Jan 3, 2025

Fix bug
XKTZ
committed
f595054
Merge remote-tracking branch 'rankllm/main' into reorder
XKTZ
committed
94a5aa5

Commits on Jan 4, 2025

update
XKTZ
committed
8d0ea17
Merge remote-tracking branch 'rankllm/main' into reorder
XKTZ
committed
ba2c3ef
Fix TensorRT
XKTZ
committed
ca0a1f0
Fix gitignore
XKTZ
committed
2efdbd5
Fix import
XKTZ
committed
dc9fdbf
Added Readme
XKTZ
committed
4669a3f
MonoT5 type def
XKTZ
committed
2a38464
Let RankFID able to fit the type def
XKTZ
committed
75c4a74

Commits on Jan 17, 2025

Remove Tournament sort method with ListT5
XKTZ
committed
0b0426d
Merge remote-tracking branch 'rankllm/main' into reorder
XKTZ
committed
fbb163c
Add a comment
XKTZ
committed
e5842a2

Commits on Jan 24, 2025

lint
XKTZ
committed
70f3914

Commits on Jan 29, 2025

Update rankfid
XKTZ
committed
4944927
Add some comments
XKTZ
committed
0d5c5e7
Merge branch 'reorder' of github.com:XKTZ/rank_llm into reorder
XKTZ
committed
8b0f963
Updated
XKTZ
committed
1115247