diff --git a/examples/baby-llama/baby-llama.cpp b/examples/baby-llama/baby-llama.cpp index bd9359801..28a40dde1 100644 --- a/examples/baby-llama/baby-llama.cpp +++ b/examples/baby-llama/baby-llama.cpp @@ -105,7 +105,7 @@ struct llama_hparams_lora { } }; -struct baby_llama_layer { +struct gpt_layer { // normalization struct ggml_tensor * attention_norm; @@ -169,7 +169,7 @@ struct llama_model { struct ggml_tensor * norm; struct ggml_tensor * output; - std::vector layers; + std::vector layers; }; struct llama_model_lora {