ejschwartz commited on
Commit
fc46b6a
·
1 Parent(s): 5767a00

Output seems truncated. Try setting max_length now that we have do_sample=True

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -66,6 +66,7 @@ def predict(code):
66
  temperature=1.2,
67
  repetition_penalty=1.1,
68
  return_full_text=False,
 
69
  )
70
  print(f"Pipe out: {repr(pipe_out)}")
71
 
 
66
  temperature=1.2,
67
  repetition_penalty=1.1,
68
  return_full_text=False,
69
+ max_length=4096,
70
  )
71
  print(f"Pipe out: {repr(pipe_out)}")
72