Update README.md
Browse files
README.md
CHANGED
@@ -59,7 +59,7 @@ AI mạnh (hay AI rộng) là hướng nghiên cứu nhằm mục đích tạo r
|
|
59 |
|
60 |
prompt = prompt.format(content, "")
|
61 |
inputs = tokenizer(prompt, return_tensors="pt")
|
62 |
-
output = model.generate(**inputs, max_new_tokens=
|
63 |
answer = tokenizer.decode(output[0], skip_special_tokens=True)
|
64 |
answer = answer.replace(prompt, "")
|
65 |
print(answer)
|
|
|
59 |
|
60 |
prompt = prompt.format(content, "")
|
61 |
inputs = tokenizer(prompt, return_tensors="pt")
|
62 |
+
output = model.generate(**inputs, max_new_tokens=2048)
|
63 |
answer = tokenizer.decode(output[0], skip_special_tokens=True)
|
64 |
answer = answer.replace(prompt, "")
|
65 |
print(answer)
|