mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-11 14:21:41 +08:00
removing unused RoPE parameters
This commit is contained in:
parent
6ea4dd3ae7
commit
a10d0ea661
@ -235,8 +235,6 @@
|
||||
" def __init__(\n",
|
||||
" self, d_in, d_out, context_length, num_heads,\n",
|
||||
" num_kv_groups,\n",
|
||||
" rope_base=10_000,\n",
|
||||
" rope_config=None,\n",
|
||||
" dtype=None\n",
|
||||
" ):\n",
|
||||
" super().__init__()\n",
|
||||
@ -325,8 +323,6 @@
|
||||
" context_length=cfg[\"context_length\"],\n",
|
||||
" num_heads=cfg[\"n_heads\"],\n",
|
||||
" num_kv_groups=cfg[\"n_kv_groups\"],\n",
|
||||
" rope_base=cfg[\"rope_base\"],\n",
|
||||
" rope_config=cfg[\"rope_freq\"],\n",
|
||||
" dtype=cfg[\"dtype\"]\n",
|
||||
" )\n",
|
||||
" self.ff = FeedForward(cfg)\n",
|
||||
|
||||
Loading…
Reference in New Issue
Block a user