From b44ada5e967120d1e369151d6a538f79639a05ec Mon Sep 17 00:00:00 2001 From: Hana Joo Date: Wed, 8 Jul 2026 02:26:13 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 944369305 --- python/convert_from_safetensors.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/convert_from_safetensors.py b/python/convert_from_safetensors.py index 11e35731..89e5e629 100644 --- a/python/convert_from_safetensors.py +++ b/python/convert_from_safetensors.py @@ -226,7 +226,7 @@ def _get_layer_config(dims: Dict[str, Any]): ], } if dims["has_post_norm"]: # See longer comment above. - config["llm-layers"] += [ + config["llm-layers"] += [ # pyrefly: ignore[bad-assignment] ( "language_model.model.layers.%d.input_layernorm.weight", (model_dim,), @@ -249,7 +249,7 @@ def _get_layer_config(dims: Dict[str, Any]): ), ] else: - config["llm-layers"] += [ + config["llm-layers"] += [ # pyrefly: ignore[bad-assignment] ( "language_model.model.layers.%d.input_layernorm.weight", (model_dim,),