akoksal commited on
Commit
6d3f295
·
1 Parent(s): 7344ca9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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
- # > </s>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,
 
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,