CreitinGameplays commited on
Commit
93d9450
1 Parent(s): 5527a29

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,8 +21,9 @@ def generate_text(prompt):
21
  num_return_sequences=1, # Generate only 1 sequence
22
  do_sample=True, # Enable sampling for creativity
23
  top_k=50, # Sample from the top 50 most likely tokens at each step
24
- top_p=0.95, # Filter out highly probable unlikely continuations
25
- temperature=1.0 # Control the randomness of the generated text (1.0 for default)
 
26
  )
27
 
28
  # Decode the generated token sequence back to text
 
21
  num_return_sequences=1, # Generate only 1 sequence
22
  do_sample=True, # Enable sampling for creativity
23
  top_k=50, # Sample from the top 50 most likely tokens at each step
24
+ top_p=0.15, # Filter out highly probable unlikely continuations
25
+ temperature=0.1 # Control the randomness of the generated text (1.0 for default)
26
+ repetition_penalty=1.165
27
  )
28
 
29
  # Decode the generated token sequence back to text