joaogante HF staff commited on
Commit
46e822a
1 Parent(s): c347b21

explicit float cast

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def run_generation(user_text, top_p, temperature, top_k, max_new_tokens):
27
  max_new_tokens=max_new_tokens,
28
  do_sample=True,
29
  top_p=top_p,
30
- temperature=temperature,
31
  top_k=top_k
32
  )
33
  t = Thread(target=model.generate, kwargs=generate_kwargs)
 
27
  max_new_tokens=max_new_tokens,
28
  do_sample=True,
29
  top_p=top_p,
30
+ temperature=float(temperature),
31
  top_k=top_k
32
  )
33
  t = Thread(target=model.generate, kwargs=generate_kwargs)