sudy-super
/

Yamase-12B

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sudy-super commited on Aug 4

Commit

087de0b

•

1 Parent(s): 1064ffd

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -48,6 +48,7 @@ with torch.no_grad():
         temperature=0.3,
         top_p=0.95,
         top_k=50,
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
     )

         temperature=0.3,
         top_p=0.95,
         top_k=50,
+        repetition_penalty=1.1,
         pad_token_id=tokenizer.pad_token_id,
         eos_token_id=tokenizer.eos_token_id,
     )