Skip to content

Commit 2989ed9

Browse files
Update lora_tuning.ipynb (#582)
Update from Keras NLP to Keras Hub
1 parent 9bac468 commit 2989ed9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

site/en/gemma/docs/core/lora_tuning.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@
319319
" instruction=\"What should I do on a trip to Europe?\",\n",
320320
" response=\"\",\n",
321321
")\n",
322-
"sampler = keras_nlp.samplers.TopKSampler(k=5, seed=2)\n",
322+
"sampler = keras_hub.samplers.TopKSampler(k=5, seed=2)\n",
323323
"gemma_lm.compile(sampler=sampler)\n",
324324
"print(gemma_lm.generate(prompt, max_length=256))"
325325
]
@@ -676,7 +676,7 @@
676676
" instruction=\"What should I do on a trip to Europe?\",\n",
677677
" response=\"\",\n",
678678
")\n",
679-
"sampler = keras_nlp.samplers.TopKSampler(k=5, seed=2)\n",
679+
"sampler = keras_hub.samplers.TopKSampler(k=5, seed=2)\n",
680680
"gemma_lm.compile(sampler=sampler)\n",
681681
"print(gemma_lm.generate(prompt, max_length=256))"
682682
]

0 commit comments

Comments
 (0)