zerank-2 / modeling_zeranker.py

Commit History

Remove unused __init__ method and related code
3aefef2

prathamj31 commited on

Load model during to_device call for eager loading
f899c80

prathamj31 commited on

Add logging statements throughout model lifecycle
456ffeb

prathamj31 commited on

Fix __init__ to call original CrossEncoder initialization
16b4531

prathamj31 commited on

Remove lazy loading and load model on instantiation
0ac03c0

prathamj31 commited on

Remove logging statements
cd705e3

wenjin_lee commited on

Use torch.inference_mode() and disable gradient checkpointing
d9aaf5e

wenjin_lee commited on

releasing zerank-2
b269504

npip99 commited on