diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 9a6c29c27bdf..e98bdd0a8da8 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -2579,7 +2579,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P if len(keys_on_cpu) > 0 and not load_in_8bit_fp32_cpu_offload: raise ValueError( "If you want to offload some keys to `cpu` or `disk`, you need to set " - "`load_in_8bit_fp32_cpu_offload=True`. Note that these modules will not be " + "`llm_int8_enable_fp32_cpu_offload=True`. Note that these modules will not be " " converted to 8-bit but kept in 32-bit." )