Unverified Commit 1ef13cf9 authored by Jee Jee Li's avatar Jee Jee Li Committed by GitHub
Browse files

[Misc]Fix BitAndBytes exception messages (#7626)

parent 832163b8
......@@ -883,11 +883,11 @@ class BitsAndBytesModelLoader(BaseModelLoader):
if not hasattr(model, 'load_weights'):
raise AttributeError(
"The required method 'load_weights' is not defined in class"
f" {type(self).__name__}.")
f" {type(model).__name__}.")
if not hasattr(model, 'bitsandbytes_stacked_params_mapping'):
raise AttributeError(
f"Model {type(self).__name__} does not support BitsAndBytes "
f"Model {type(model).__name__} does not support BitsAndBytes "
"quantization yet.")
logger.info("Loading weights with BitsAndBytes quantization. "
......
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