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))
|