Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ peft_model.config.max_length=512
|
|
| 58 |
peft_tokenizer.model_max_length=512
|
| 59 |
inputs = peft_tokenizer(inputs, return_tensors="pt")
|
| 60 |
outputs = peft_model.generate(**inputs,max_length=512)
|
| 61 |
-
answer =
|
| 62 |
from textwrap import fill
|
| 63 |
|
| 64 |
print(fill(answer, width=80))
|
|
|
|
| 58 |
peft_tokenizer.model_max_length=512
|
| 59 |
inputs = peft_tokenizer(inputs, return_tensors="pt")
|
| 60 |
outputs = peft_model.generate(**inputs,max_length=512)
|
| 61 |
+
answer = peft_tokenizer.decode(outputs[0])
|
| 62 |
from textwrap import fill
|
| 63 |
|
| 64 |
print(fill(answer, width=80))
|