waddledee
/

three-line-summarization-ja

Model card Files Files and versions Community

waddledee commited on May 28

Commit

692c679

•

1 Parent(s): e04ff45

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -52,7 +52,7 @@ def gen(text, model):
 入力文:
 {text} [/INST] [R_START]
 """
-    token_ids = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
     token_ids.to("cuda")
     with torch.no_grad():

 入力文:
 {text} [/INST] [R_START]
 """
+    token_ids = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt", truncation=True, max_length=4096)
     token_ids.to("cuda")
     with torch.no_grad():