naver
/

trecdl22-crossencoder-rankT53b-repro

Model card Files Files and versions Community

cadurosar commited on Jul 10, 2023

Commit

ff4e33d

·

1 Parent(s): d32f7fb

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -47,9 +47,11 @@ original_model="t5-3b"
 path_checkpoint="trecdl22-crossencoder-rankT53b-repro/pytorch_model.bin"
 print("Loading")
-model = T5EncoderRerank(original_model,bf16=True)
 model.load_state_dict(torch.load(path_checkpoint,map_location=torch.device("cpu")))
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 model.to(device)
 tokenizer = AutoTokenizer.from_pretrained(original_model)
 ```

 path_checkpoint="trecdl22-crossencoder-rankT53b-repro/pytorch_model.bin"
 print("Loading")
+model = T5EncoderRerank(original_model)
 model.load_state_dict(torch.load(path_checkpoint,map_location=torch.device("cpu")))
 device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
 model.to(device)
 tokenizer = AutoTokenizer.from_pretrained(original_model)
+print("loaded")
 ```