Skip to content

Commit fa494fa

Browse files
committed
vocab size should be 50257 not 50256
1 parent 664c633 commit fa494fa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

ch07/04_preference-tuning-with-dpo/dpo-from-scratch.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2140,7 +2140,7 @@
21402140
},
21412141
"source": [
21422142
"- In other words, `torch.gather` is a selection function\n",
2143-
"- When we computed the loss earlier, we used it to retrieve the log probabilities corresponding to the correct token in the 50,256-token vocabulary\n",
2143+
"- When we computed the loss earlier, we used it to retrieve the log probabilities corresponding to the correct token in the 50,257-token vocabulary\n",
21442144
"- The \"correct\" tokens are the tokens given in the response entry"
21452145
]
21462146
},
@@ -3112,7 +3112,7 @@
31123112
"provenance": []
31133113
},
31143114
"kernelspec": {
3115-
"display_name": ".venv",
3115+
"display_name": "Python 3 (ipykernel)",
31163116
"language": "python",
31173117
"name": "python3"
31183118
},
@@ -3126,7 +3126,7 @@
31263126
"name": "python",
31273127
"nbconvert_exporter": "python",
31283128
"pygments_lexer": "ipython3",
3129-
"version": "3.12.6"
3129+
"version": "3.10.16"
31303130
}
31313131
},
31323132
"nbformat": 4,

0 commit comments

Comments
 (0)