Skip to content

Commit de6394a

Browse files
committed
better message when keep_in_fp32_modules.
1 parent ff8ddef commit de6394a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -701,7 +701,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
701701
low_cpu_mem_usage = True
702702
logger.info("Set `low_cpu_mem_usage` to True as `_keep_in_fp32_modules` is not None.")
703703
elif not low_cpu_mem_usage:
704-
raise ValueError("`low_cpu_mem_usage` cannot be False or None when using quantization.")
704+
raise ValueError("`low_cpu_mem_usage` cannot be False when `keep_in_fp32_modules` is True.")
705705
else:
706706
keep_in_fp32_modules = []
707707
#######################################

0 commit comments

Comments
 (0)