nicholasKluge
/

Aira-2-355M

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

db6246b

·

1 Parent(s): 29dd0b5

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -85,7 +85,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,