pnpm12 commited on
Commit
57d7fea
1 Parent(s): 073953c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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=1024)
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)