Skip to content

Commit

Permalink
Fix pip install --upgrade accelerate command in modeling_utils.py (h…
Browse files Browse the repository at this point in the history
…uggingface#23747)

Fix command in modeling_utils.py
  • Loading branch information
tloen authored and gojiteji committed Jun 5, 2023
1 parent a6668e1 commit 67b8604
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/modeling_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -2727,7 +2727,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
raise ValueError(
"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,"
"`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"
)
elif load_in_8bit:
Expand Down

0 comments on commit 67b8604

Please sign in to comment.