Merge remote-tracking branch 'origin' into kylesayrs/reduce-quantized-compression-memory

kylesayrs · kylesayrs · commit 1a70148663d2 · 2025-05-07T21:38:51.000-04:00
diff --git a/src/compressed_tensors/utils/safetensors_load.py b/src/compressed_tensors/utils/safetensors_load.py
@@ -250,7 +250,7 @@ def get_nested_weight_mappings(
 
 
 def get_nested_mappings_from_state_dict(
-    state_dict,
+    state_dict: Dict[str, Tensor],
     params_to_nest: Iterable[str],
     return_unmatched_params: bool = False,
 ) -> Union[NestedStateDictType, Tuple[NestedStateDictType, Dict[str, Tensor]]]:
