From b5ef6f29b35777255e80123d5bd6079c2778c3cc Mon Sep 17 00:00:00 2001 From: Divyashree Sreepathihalli Date: Tue, 22 Oct 2024 13:35:28 -0700 Subject: [PATCH] Update llama_backbone.py docstring --- keras_hub/src/models/llama/llama_backbone.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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.