Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -76,11 +76,10 @@ def stream_chat(
|
|
76 |
|
77 |
Importantly, always respond to and rebut the previous speaker's points in Trump's style. Keep responses concise and avoid unnecessary repetition."""
|
78 |
|
79 |
-
temperature = 0.
|
80 |
-
max_new_tokens =
|
81 |
-
top_p = 0
|
82 |
-
top_k =
|
83 |
-
repetition_penalty = 1.5
|
84 |
|
85 |
conversation = [
|
86 |
{"role": "system", "content": system_prompt}
|
@@ -108,7 +107,6 @@ Importantly, always respond to and rebut the previous speaker's points in Trump'
|
|
108 |
pad_token_id=tokenizer.pad_token_id,
|
109 |
eos_token_id=tokenizer.eos_token_id,
|
110 |
streamer=streamer,
|
111 |
-
repetition_penalty=repetition_penalty,
|
112 |
)
|
113 |
|
114 |
with torch.no_grad():
|
|
|
76 |
|
77 |
Importantly, always respond to and rebut the previous speaker's points in Trump's style. Keep responses concise and avoid unnecessary repetition."""
|
78 |
|
79 |
+
temperature = 0.8
|
80 |
+
max_new_tokens = 1024
|
81 |
+
top_p = 1.0
|
82 |
+
top_k = 20
|
|
|
83 |
|
84 |
conversation = [
|
85 |
{"role": "system", "content": system_prompt}
|
|
|
107 |
pad_token_id=tokenizer.pad_token_id,
|
108 |
eos_token_id=tokenizer.eos_token_id,
|
109 |
streamer=streamer,
|
|
|
110 |
)
|
111 |
|
112 |
with torch.no_grad():
|