Update src/diffusers/models/modeling_utils.py

a-r-r-o-w · sayakpaul · web-flow · commit 9f98833cfba5 · 2024-12-25T09:54:17.000+05:30
Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;
diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py
@@ -819,6 +819,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
                     revision=revision,
                     subfolder=subfolder or "",
                 )
+                # TODO: https://github.com/huggingface/diffusers/issues/10013
                 if hf_quantizer is not None:
                     model_file = _merge_sharded_checkpoints(sharded_ckpt_cached_folder, sharded_metadata)
                     logger.info("Merged sharded checkpoints as `hf_quantizer` is not None.")

Original file line number	Diff line number	Diff line change
`@@ -819,6 +819,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P`
`819`	`819`	`revision=revision,`
`820`	`820`	`subfolder=subfolder or "",`
`821`	`821`	`)`
	`822`	`+ # TODO: https://github.com/huggingface/diffusers/issues/10013`
`822`	`823`	`if hf_quantizer is not None:`
`823`	`824`	`model_file = _merge_sharded_checkpoints(sharded_ckpt_cached_folder, sharded_metadata)`
`824`	`825`	logger.info("Merged sharded checkpoints as `hf_quantizer` is not None.")