castorini
/

rankllama-v1-7b-lora-doc

Model card Files Files and versions Community

MrLight commited on Oct 23, 2023

Commit

2435db0

•

1 Parent(s): 3d1e5df

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -41,7 +41,7 @@ title = "Llama"
 document = "The llama is a domesticated South American camelid, widely used as a meat and pack animal by Andean cultures since the pre-Columbian era."
 # Tokenize the query-document pair
-inputs = tokenizer(f'query: {query}', f'document: {url} {title} {document}</s>', return_tensors='pt')
 # Run the model forward
 with torch.no_grad():

 document = "The llama is a domesticated South American camelid, widely used as a meat and pack animal by Andean cultures since the pre-Columbian era."
 # Tokenize the query-document pair
+inputs = tokenizer(f'query: {query}', f'document: {url} {title} {document}', return_tensors='pt')
 # Run the model forward
 with torch.no_grad():