From 9636787101ce4b8be0a05d3d1cf850b196e74bea Mon Sep 17 00:00:00 2001 From: Bryce Meyer Date: Tue, 31 Dec 2024 02:22:58 +0100 Subject: [PATCH] ran format --- transformer_lens/loading_from_pretrained.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformer_lens/loading_from_pretrained.py b/transformer_lens/loading_from_pretrained.py index 77bc64ae1..245b792fc 100644 --- a/transformer_lens/loading_from_pretrained.py +++ b/transformer_lens/loading_from_pretrained.py @@ -757,7 +757,7 @@ def convert_hf_model_config(model_name: str, **kwargs): ) architecture = hf_config.architectures[0] - print ("model name", official_model_name) + print("model name", official_model_name) if official_model_name.startswith( ("llama-7b", "meta-llama/Llama-2-7b") ): # same architecture for LLaMA and Llama-2