diff --git a/keras_hub/src/models/llama/llama_backbone.py b/keras_hub/src/models/llama/llama_backbone.py index a9d5622305..0e923c29cd 100644 --- a/keras_hub/src/models/llama/llama_backbone.py +++ b/keras_hub/src/models/llama/llama_backbone.py @@ -59,7 +59,7 @@ class LlamaBackbone(Backbone): } # Pretrained Llama decoder. - model = keras_hub.models.LlamaBackbone.from_preset("llama7b_base_en") + model = keras_hub.models.LlamaBackbone.from_preset("llama2_7b_en") model(input_data) # Randomly initialized Llama decoder with custom config.