formated code with black

TransformerLensOrg · Oct 2, 2024 · cf40f5b · cf40f5b
1 parent a075e4b
commit cf40f5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_lens/loading_from_pretrained.py b/transformer_lens/loading_from_pretrained.py
@@ -1201,7 +1201,7 @@ def convert_hf_model_config(model_name: str, **kwargs):
             "rotary_dim": hf_config.hidden_size // hf_config.num_attention_heads,
             "final_rms": True,
             "gated_mlp": True,
-        }       
+        }
     elif architecture == "QWenLMHeadModel":
         cfg_dict = {
             "d_model": hf_config.hidden_size,