Skip to content

Commit

Permalink
Fix typo at ImportError (#30090)
Browse files Browse the repository at this point in the history
fix typo at ImportError
  • Loading branch information
DrAnaximandre authored Apr 8, 2024
1 parent ef38e2a commit 7afade2
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/transformers/quantizers/quantizer_quanto.py
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,9 @@ def validate_environment(self, *args, **kwargs):
if not is_quanto_available():
raise ImportError("Loading a quanto quantized model requires quanto library (`pip install quanto`)")
if not is_accelerate_available():
raise ImportError("Loading a quanto quantized model requires accelerate library (`pip install quanto`)")
raise ImportError(
"Loading a quanto quantized model requires accelerate library (`pip install accelerate`)"
)

def update_device_map(self, device_map):
if device_map is None:
Expand Down

0 comments on commit 7afade2

Please sign in to comment.