Update README.md
Browse files
README.md
CHANGED
@@ -42,5 +42,5 @@ instruction = "Write a function to scrape hacker news."
|
|
42 |
prompt = f"Below is an instruction that describes a task.\n Write a response that appropriately completes the request.\n\n ### Instruction:\n{instruction}\n\n### Response:"
|
43 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
44 |
outputs = model.generate(**inputs,temperature=0.3,do_sample=True)
|
45 |
-
print(tokenizer.decode(outputs[0],skip_special_tokens=True))
|
46 |
```
|
|
|
42 |
prompt = f"Below is an instruction that describes a task.\n Write a response that appropriately completes the request.\n\n ### Instruction:\n{instruction}\n\n### Response:"
|
43 |
inputs = tokenizer(prompt, return_tensors="pt").to(device)
|
44 |
outputs = model.generate(**inputs,temperature=0.3,do_sample=True)
|
45 |
+
print(tokenizer.decode(outputs[0],skip_special_tokens=True,max_new_tokens=200))
|
46 |
```
|