Update README.md
Browse files
README.md
CHANGED
@@ -57,7 +57,12 @@ inputs = processor.apply_chat_template(
|
|
57 |
|
58 |
# autoregressively complete prompt
|
59 |
output = model.generate(**inputs, max_new_tokens=100)
|
60 |
-
print(processor.decode(output[0], skip_special_tokens=True))
|
|
|
|
|
|
|
|
|
|
|
61 |
```
|
62 |
|
63 |
### Donation
|
|
|
57 |
|
58 |
# autoregressively complete prompt
|
59 |
output = model.generate(**inputs, max_new_tokens=100)
|
60 |
+
# print(processor.decode(output[0], skip_special_tokens=True))
|
61 |
+
|
62 |
+
response = processor.decode(output[0], skip_special_tokens=True)
|
63 |
+
cleaned_response = response.split("<|assistant|>")[1].split("<")[0].strip()
|
64 |
+
|
65 |
+
print(cleaned_response)
|
66 |
```
|
67 |
|
68 |
### Donation
|