arcee-ai
/

Trinity-Large-Preview

Text Generation

Model card Files Files and versions

bartowski commited on 27 days ago

Commit

be9a3f8

·

verified ·

1 Parent(s): 29f8dd2

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -147,9 +147,9 @@ outputs = model.generate(
     input_ids,
     max_new_tokens=256,
     do_sample=True,
-    temperature=0.5,
     top_k=50,
-    top_p=0.95
 )
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)

     input_ids,
     max_new_tokens=256,
     do_sample=True,
+    temperature=0.8,
     top_k=50,
+    top_p=0.8
 )
 response = tokenizer.decode(outputs[0], skip_special_tokens=True)