Compare commits

..

1 Commits

Author SHA1 Message Date
Jeff Wainwright
f3a3db1881
Merge e0c6b59d39 into ec0a832acb 2026-01-09 01:52:09 -08:00

View File

@ -546,8 +546,7 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
weight_key = f"{prefix}weight"
weight = state_dict.pop(weight_key, None)
if weight is None:
logging.warning(f"Missing weight for layer {layer_name}")
return
raise ValueError(f"Missing weight for layer {layer_name}")
manually_loaded_keys = [weight_key]