Update llama_backbone.py docstring (#1950)

keras-team · Oct 22, 2024 · 3b331dd · 3b331dd
1 parent ffff5c3
commit 3b331dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.