Update You_can_still_run_it_directly_on_your_own_machine_if_it's_too_small.ipynb
Browse files
You_can_still_run_it_directly_on_your_own_machine_if_it's_too_small.ipynb
CHANGED
|
@@ -188,7 +188,7 @@
|
|
| 188 |
"outputs": [],
|
| 189 |
"source": [
|
| 190 |
"@torch.no_grad()\n",
|
| 191 |
-
"def chat(user_input, max_new_tokens=200, temperature=0.
|
| 192 |
" prompt = f\"{USER_TOK}{user_input}{ASSISTANT_TOK}\"\n",
|
| 193 |
" prompt_ids = tokenizer_backend.encode(prompt).ids\n",
|
| 194 |
" prompt_ids = prompt_ids[-model.block_size:]\n",
|
|
|
|
| 188 |
"outputs": [],
|
| 189 |
"source": [
|
| 190 |
"@torch.no_grad()\n",
|
| 191 |
+
"def chat(user_input, max_new_tokens=200, temperature=0.5, top_k=10):\n",
|
| 192 |
" prompt = f\"{USER_TOK}{user_input}{ASSISTANT_TOK}\"\n",
|
| 193 |
" prompt_ids = tokenizer_backend.encode(prompt).ids\n",
|
| 194 |
" prompt_ids = prompt_ids[-model.block_size:]\n",
|