castorini
/

rankllama-v1-7b-lora-passage

Model card Files Files and versions Community

MrLight commited on Oct 23, 2023

Commit

d8b8aad

·

1 Parent(s): 2b74e56

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -42,7 +42,7 @@ title = "Llama"
 passage = "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-passage pair
-inputs = tokenizer(f'query: {query}', f'document: {title} {passage}</s>', return_tensors='pt')
 # Run the model forward
 with torch.no_grad():

 passage = "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-passage pair
+inputs = tokenizer(f'query: {query}', f'document: {title} {passage}', return_tensors='pt')
 # Run the model forward
 with torch.no_grad():