Unverified Commit f0a2a82a authored by Eric J. Wang's avatar Eric J. Wang Committed by GitHub
Browse files

Fix `pip install --upgrade accelerate` command in modeling_utils.py (#23747)

Fix command in modeling_utils.py
parent e45e756d
...@@ -2727,7 +2727,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -2727,7 +2727,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
raise ValueError( raise ValueError(
"You are using `device_map='auto'` on a 4bit loaded version of the model. To automatically compute" "You are using `device_map='auto'` on a 4bit loaded version of the model. To automatically compute"
" the appropriate device map, you should upgrade your `accelerate` library," " the appropriate device map, you should upgrade your `accelerate` library,"
"`pip install --upgrade accelerare` or install it from source to support fp4 auto device map" "`pip install --upgrade accelerate` or install it from source to support fp4 auto device map"
"calculation. You may encounter unexpected behavior, or pass your own device map" "calculation. You may encounter unexpected behavior, or pass your own device map"
) )
elif load_in_8bit: elif load_in_8bit:
......
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