尝试在 Kaggle 上运行 Gemma 并遇到问题“keras_nlp.models”没有属性“GemmaCausalLM”

Question

我正在尝试使用此模型在 Keras 上运行 Gemma https://www.kaggle.com/models/keras/gemma/frameworks/keras/variations/gemma_instruct_7b_en 我正在复制上页“模型卡”上提供的示例。 当我运行这段代码时：

gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("gemma_instruct_7b_en")
gemma_lm.generate("Keras is a", max_length=30)

# Generate with batched prompts.
gemma_lm.generate(["Keras is a", "I want to say"], max_length=30)

我收到此错误：

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
Cell In[2], line 1
----> 1 gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("gemma_instruct_7b_en")
      2 gemma_lm.generate("Keras is a", max_length=30)
      4 # Generate with batched prompts.
AttributeError: module 'keras_nlp.models' has no attribute 'GemmaCausalLM'

我该如何解决这个问题？

Answer 1

我对 Gemma 2B 型号也有同样的问题： https://www.kaggle.com/models/keras/gemma/frameworks/keras/variations/gemma_2b_en/versions/2

gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("gemma_2b_en")

因相同的 AttributeError 而失败

AttributeError: module 'keras_nlp.models' has no attribute 'GemmaCausalLM'

我检查了 Keras 文档，但他们在以下位置提供了相同的代码行 https://keras.io/api/keras_nlp/models/gemma/gemma_causal_lm/

所以看起来要么是 Kaggle 的短暂故障，要么是 Keras 的未记录的更改