cyberagent
/

open-calm-medium

Text Generation

text-generation-inference

Model card Files Files and versions

rishigami commited on May 18, 2023

Commit

0b7045d

·

1 Parent(s): 73279e7

Update README.md

Files changed (1) hide show

README.md +2 -0

README.md CHANGED Viewed

@@ -33,6 +33,8 @@ with torch.no_grad():
         max_new_tokens=64,
         do_sample=True,
         temperature=0.7,
         pad_token_id=tokenizer.pad_token_id,
     )

         max_new_tokens=64,
         do_sample=True,
         temperature=0.7,
+        top_p=0.9,
+        repetition_penalty=1.05,
         pad_token_id=tokenizer.pad_token_id,
     )