Danielrahmai1991 commited on
Commit
97e04f5
1 Parent(s): dc1fe45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def llm_run(prompt, max_length, top_p, temprature, top_k, messages):
66
  )
67
  generate_kwargs = dict(
68
  max_length=int(max_length),top_p=float(top_p), do_sample=True,
69
- top_k=int(top_k), streamer=streamer, temperature=int(temprature), repetition_penalty=1.2
70
  )
71
 
72
  t = Thread(target=model.generate, args=(input_ids,), kwargs=generate_kwargs)
 
66
  )
67
  generate_kwargs = dict(
68
  max_length=int(max_length),top_p=float(top_p), do_sample=True,
69
+ top_k=int(top_k), streamer=streamer, temperature=float(temprature), repetition_penalty=1.2
70
  )
71
 
72
  t = Thread(target=model.generate, args=(input_ids,), kwargs=generate_kwargs)