Commit History
Load model during to_device call for eager loading
f899c80
Add logging statements throughout model lifecycle
456ffeb
Fix __init__ to call original CrossEncoder initialization
16b4531
Remove lazy loading and load model on instantiation
0ac03c0
Remove logging statements
cd705e3
wenjin_lee
commited on
Use torch.inference_mode() and disable gradient checkpointing
d9aaf5e
wenjin_lee
commited on
