From 0bfcf1c062c824204c1b1b41fc5d7bf138d4f282 Mon Sep 17 00:00:00 2001 From: Ahmed Ahmed Date: Wed, 11 Sep 2024 12:19:20 -0700 Subject: [PATCH] new config --- src/levanter/models/llama.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/levanter/models/llama.py b/src/levanter/models/llama.py index 2a2d2664d..94b01f2c3 100644 --- a/src/levanter/models/llama.py +++ b/src/levanter/models/llama.py @@ -109,6 +109,9 @@ def hf_checkpoint_converter(self) -> HFCheckpointConverter["LlamaConfig"]: # ty @classmethod def from_hf_config(cls, hf_config: HfConfig): + print(f'\n hf_config: {hf_config}') + print(type(hf_config)) + print(dir(hf_config)) return LlamaConfig( seq_len=hf_config.max_position_embeddings, hidden_dim=hf_config.hidden_size,