Spaces:
Sleeping
Sleeping
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)
|