MatteoScript commited on
Commit
f21499d
·
1 Parent(s): a061413

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
- stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
41
 
42
  # Accumula l'output in una lista
43
  output_list = []
 
37
  seed=42,
38
  )
39
  formatted_prompt = format_prompt(prompt, history)
40
+ stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
41
 
42
  # Accumula l'output in una lista
43
  output_list = []