From 3b331ddfbe9a7ebdc304c171c09fb46b48a4406b Mon Sep 17 00:00:00 2001
From: Divyashree Sreepathihalli <divyashreepathihalli@gmail.com>
Date: Tue, 22 Oct 2024 13:40:48 -0700
Subject: [PATCH] Update llama_backbone.py docstring (#1950)
---
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.