From e42368e62c6a844735dc6998900b5759627f7ee9 Mon Sep 17 00:00:00 2001 From: Sachin Prasad Date: Tue, 6 May 2025 11:04:50 -0700 Subject: [PATCH] Update lora_tuning.ipynb Update from Keras NLP to Keras Hub --- site/en/gemma/docs/core/lora_tuning.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/site/en/gemma/docs/core/lora_tuning.ipynb b/site/en/gemma/docs/core/lora_tuning.ipynb index 391f25f8b..e34ccb79f 100644 --- a/site/en/gemma/docs/core/lora_tuning.ipynb +++ b/site/en/gemma/docs/core/lora_tuning.ipynb @@ -319,7 +319,7 @@ " instruction=\"What should I do on a trip to Europe?\",\n", " response=\"\",\n", ")\n", - "sampler = keras_nlp.samplers.TopKSampler(k=5, seed=2)\n", + "sampler = keras_hub.samplers.TopKSampler(k=5, seed=2)\n", "gemma_lm.compile(sampler=sampler)\n", "print(gemma_lm.generate(prompt, max_length=256))" ] @@ -676,7 +676,7 @@ " instruction=\"What should I do on a trip to Europe?\",\n", " response=\"\",\n", ")\n", - "sampler = keras_nlp.samplers.TopKSampler(k=5, seed=2)\n", + "sampler = keras_hub.samplers.TopKSampler(k=5, seed=2)\n", "gemma_lm.compile(sampler=sampler)\n", "print(gemma_lm.generate(prompt, max_length=256))" ]