Skip to content

Commit c78dd0c

Browse files
committed
raise an error for non-True low_cpu_mem_usage values when using quant.
1 parent 5e722cd commit c78dd0c

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/diffusers/models/modeling_utils.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -683,9 +683,8 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
683683
user_agent["quant"] = hf_quantizer.quantization_config.quant_method.value
684684

685685
# Force-set to `True` for more mem efficiency
686-
if low_cpu_mem_usage is None:
687-
low_cpu_mem_usage = True
688-
logger.warning("`low_cpu_mem_usage` was None, now set to True since model is quantized.")
686+
if not low_cpu_mem_usage:
687+
raise ValueError("`low_cpu_mem_usage` cannot be False or None when using quantization.")
689688

690689
# Check if `_keep_in_fp32_modules` is not None
691690
use_keep_in_fp32_modules = (cls._keep_in_fp32_modules is not None) and (

0 commit comments

Comments
 (0)