MatteoScript commited on
Commit
609a4fb
·
1 Parent(s): f381f25

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -37,7 +37,7 @@ def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95,
37
  seed=42,
38
  )
39
  formatted_prompt = format_prompt(prompt, history)
40
- output = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True)
41
  return output
42
 
43
  #stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=False, return_full_text=False)
 
37
  seed=42,
38
  )
39
  formatted_prompt = format_prompt(prompt, history)
40
+ output = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False)
41
  return output
42
 
43
  #stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=False, return_full_text=False)