mondk commited on
Commit
ec4a52d
·
verified ·
1 Parent(s): 6620b62

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.8, top_k=40):\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",
 
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",