unicamp-dl
/

monoptt5-3b

Text2Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

marcospiau commited on Jun 16

Commit

ab6e00a

•

1 Parent(s): 494b63b

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -27,7 +27,7 @@ docs = [
 ranker = Reranker(
     "unicamp-dl/monoptt5-3b",
     inputs_template="Pergunta: {query} Documento: {text} Relevante:",
-    dtype=torch.float32 ## or bfloat16 if supported by your GPU
 )
 results = ranker.rank(query, docs)

 ranker = Reranker(
     "unicamp-dl/monoptt5-3b",
     inputs_template="Pergunta: {query} Documento: {text} Relevante:",
+    dtype=torch.float32 # or bfloat16 if supported by your GPU
 )
 results = ranker.rank(query, docs)