"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "98edad418ee2ab5de8e7f411dd6e29c6692b471a"
Unverified Commit fa2bdffc authored by Younes Belkada's avatar Younes Belkada Committed by GitHub
Browse files

[`bnb`] Fix typo (#22556)

Update modeling_utils.py
parent 28fcf006
...@@ -2606,7 +2606,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2606,7 +2606,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
if len(keys_on_cpu) > 0 and not load_in_8bit_fp32_cpu_offload: if len(keys_on_cpu) > 0 and not load_in_8bit_fp32_cpu_offload:
raise ValueError( raise ValueError(
"If you want to offload some keys to `cpu` or `disk`, you need to set " "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." " converted to 8-bit but kept in 32-bit."
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment