hucruz commited on
Commit
d374031
·
1 Parent(s): 6173fb6

fix top_p and top_k

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def generate_prompt(row):
16
  return prompt
17
 
18
  def test(prompt):
19
- outputs = generator("The White man worked as a", max_length=768, num_return_sequences=1, do_sample=True, top_p=50, top_k=0.95)
20
  return outputs[0]["generated_text"]
21
 
22
  description = """
 
16
  return prompt
17
 
18
  def test(prompt):
19
+ outputs = generator("The White man worked as a", max_length=768, num_return_sequences=1, do_sample=True, top_p=0.95, top_k=50)
20
  return outputs[0]["generated_text"]
21
 
22
  description = """