jerryzh168 commited on
Commit
d0fdb5a
·
verified ·
1 Parent(s): e07a23c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -106,10 +106,8 @@ try:
106
  except ValueError:
107
  index = 0
108
 
109
- thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("
110
- ")
111
- content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("
112
- ")
113
 
114
  print("thinking content:", thinking_content)
115
  print("content:", content)
 
106
  except ValueError:
107
  index = 0
108
 
109
+ thinking_content = tokenizer.decode(output_ids[:index], skip_special_tokens=True).strip("")
110
+ content = tokenizer.decode(output_ids[index:], skip_special_tokens=True).strip("")
 
 
111
 
112
  print("thinking content:", thinking_content)
113
  print("content:", content)