diff --git a/vllm/model_executor/models/utils.py b/vllm/model_executor/models/utils.py index c6498319cd..bbb7a54b55 100644 --- a/vllm/model_executor/models/utils.py +++ b/vllm/model_executor/models/utils.py @@ -321,6 +321,14 @@ class AutoWeightsLoader: f"into a single parameter {base_prefix!r}" ) + if param.dtype != weight_data.dtype: + logger.warning( + "Attempted to load weight %s with dtype %s into " + "parameter with dtype %s", + weight_qualname, + weight_data.dtype, + param.dtype, + ) weight_loader = getattr(param, "weight_loader", default_weight_loader) weight_loader(param, weight_data)