nicholasKluge
/

Aira-2-1B5

Text Generation

instruction tuned

text generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

nicholasKluge commited on Dec 8, 2023

Commit

39fedf1

•

1 Parent(s): ac9fca3

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -79,7 +79,9 @@ aira.to(device)
 question =  input("Enter your question: ")
-inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token, return_tensors="pt").to(device)
 responses = aira.generate(**inputs,
 	do_sample=True,

 question =  input("Enter your question: ")
+inputs = tokenizer(tokenizer.bos_token + question + tokenizer.sep_token,
+  add_special_tokens=False,
+  return_tensors="pt").to(device)
 responses = aira.generate(**inputs,
 	do_sample=True,