francislabounty
commited on
Commit
•
03c7d6f
1
Parent(s):
c6d1699
Update README.md
Browse files
README.md
CHANGED
@@ -53,4 +53,5 @@ prompt = construct_prompt(messages)
|
|
53 |
inputs = tokenizer(prompt, return_tensors="pt")
|
54 |
inputs = inputs.to(model.device)
|
55 |
pred = model.generate(**inputs, max_length=4096, do_sample=True, top_k=50, top_p=0.99, temperature=0.9, num_return_sequences=1)
|
56 |
-
print(tokenizer.decode(pred.cpu()[0], skip_special_tokens=True))
|
|
|
|
53 |
inputs = tokenizer(prompt, return_tensors="pt")
|
54 |
inputs = inputs.to(model.device)
|
55 |
pred = model.generate(**inputs, max_length=4096, do_sample=True, top_k=50, top_p=0.99, temperature=0.9, num_return_sequences=1)
|
56 |
+
print(tokenizer.decode(pred.cpu()[0], skip_special_tokens=True))
|
57 |
+
```
|