Update README.md
Browse files
README.md
CHANGED
@@ -63,9 +63,9 @@ instruction = "Write an essay about meditation. [EOI]"
|
|
63 |
torch.manual_seed(42)
|
64 |
input_ids = tokenizer(instruction, return_tensors="pt").input_ids
|
65 |
target_ids = model.generate(input_ids, do_sample=True, max_new_tokens=50, top_p=0.9)
|
66 |
-
tokenizer.decode(target_ids[0])
|
67 |
# Output:
|
68 |
-
# >
|
69 |
# mind, focus on the present, and achieve a calm and serene state of mind.\
|
70 |
# In today's society, people are constantly running from one thing to the next.\
|
71 |
# In order to maintain good mental health,
|
|
|
63 |
torch.manual_seed(42)
|
64 |
input_ids = tokenizer(instruction, return_tensors="pt").input_ids
|
65 |
target_ids = model.generate(input_ids, do_sample=True, max_new_tokens=50, top_p=0.9)
|
66 |
+
tokenizer.decode(target_ids[0], skip_special_tokens=True)
|
67 |
# Output:
|
68 |
+
# > Write an essay about meditation. [EOI]Meditation is a way to quiet the\
|
69 |
# mind, focus on the present, and achieve a calm and serene state of mind.\
|
70 |
# In today's society, people are constantly running from one thing to the next.\
|
71 |
# In order to maintain good mental health,
|