Update README.md
Browse files
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)
|