Spaces:
Running
Running
Commit
·
609a4fb
1
Parent(s):
f381f25
Update main.py
Browse files
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=
|
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)
|