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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -40,7 +40,7 @@ def generate(prompt, history, temperature=0.2, max_new_tokens=30000, top_p=0.95,
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=True, return_full_text=False)
44
  # Accumula l'output in una lista
45
  #output_list = []
46
  #for response in stream:
 
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)
44
  # Accumula l'output in una lista
45
  #output_list = []
46
  #for response in stream: