diffusers 0.32.0__py3-none-any.whl → 0.32.1__py3-none-any.whl

diffusers/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.32.0"
+__version__ = "0.32.1"
 
 from typing import TYPE_CHECKING
 

diffusers/models/modeling_utils.py

@@ -718,10 +718,9 @@ class ModelMixin(torch.nn.Module, PushToHubMixin):
             hf_quantizer = None
 
         if hf_quantizer is not None:
-            is_bnb_quantization_method = hf_quantizer.quantization_config.quant_method.value == "bitsandbytes"
-            if is_bnb_quantization_method and device_map is not None:
+            if device_map is not None:
                 raise NotImplementedError(
-                    "Currently, `device_map` is automatically inferred for quantized bitsandbytes models. Support for providing `device_map` as an input will be added in the future."
+                    "Currently, providing `device_map` is not supported for quantized models. Providing `device_map` as an input will be added in the future."
                 )
 
             hf_quantizer.validate_environment(torch_dtype=torch_dtype, from_flax=from_flax, device_map=device_map)
@@ -820,7 +819,8 @@ class ModelMixin(torch.nn.Module, PushToHubMixin):
                     revision=revision,
                     subfolder=subfolder or "",
                 )
-                if hf_quantizer is not None and is_bnb_quantization_method:
+                # 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.")
                     is_sharded = False

diffusers/quantizers/torchao/torchao_quantizer.py

@@ -23,7 +23,7 @@ from typing import TYPE_CHECKING, Any, Dict, List, Union
 
 from packaging import version
 
-from ...utils import get_module_from_name, is_torch_available, is_torchao_available, logging
+from ...utils import get_module_from_name, is_torch_available, is_torch_version, is_torchao_available, logging
 from ..base import DiffusersQuantizer
 
 
@@ -35,21 +35,28 @@ if is_torch_available():
     import torch
     import torch.nn as nn
 
-    SUPPORTED_TORCH_DTYPES_FOR_QUANTIZATION = (
-        # At the moment, only int8 is supported for integer quantization dtypes.
-        # In Torch 2.6, int1-int7 will be introduced, so this can be visited in the future
-        # to support more quantization methods, such as intx_weight_only.
-        torch.int8,
-        torch.float8_e4m3fn,
-        torch.float8_e5m2,
-        torch.uint1,
-        torch.uint2,
-        torch.uint3,
-        torch.uint4,
-        torch.uint5,
-        torch.uint6,
-        torch.uint7,
-    )
+    if is_torch_version(">=", "2.5"):
+        SUPPORTED_TORCH_DTYPES_FOR_QUANTIZATION = (
+            # At the moment, only int8 is supported for integer quantization dtypes.
+            # In Torch 2.6, int1-int7 will be introduced, so this can be visited in the future
+            # to support more quantization methods, such as intx_weight_only.
+            torch.int8,
+            torch.float8_e4m3fn,
+            torch.float8_e5m2,
+            torch.uint1,
+            torch.uint2,
+            torch.uint3,
+            torch.uint4,
+            torch.uint5,
+            torch.uint6,
+            torch.uint7,
+        )
+    else:
+        SUPPORTED_TORCH_DTYPES_FOR_QUANTIZATION = (
+            torch.int8,
+            torch.float8_e4m3fn,
+            torch.float8_e5m2,
+        )
 
 if is_torchao_available():
     from torchao.quantization import quantize_
@@ -125,7 +132,7 @@ class TorchAoHfQuantizer(DiffusersQuantizer):
     def update_torch_dtype(self, torch_dtype):
         quant_type = self.quantization_config.quant_type
 
-        if quant_type.startswith("int"):
+        if quant_type.startswith("int") or quant_type.startswith("uint"):
             if torch_dtype is not None and torch_dtype != torch.bfloat16:
                 logger.warning(
                     f"You are trying to set torch_dtype to {torch_dtype} for int4/int8/uintx quantization, but "

{diffusers-0.32.0.dist-info → diffusers-0.32.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: diffusers
-Version: 0.32.0
+Version: 0.32.1
 Summary: State-of-the-art diffusion in PyTorch and JAX.
 Home-page: https://github.com/huggingface/diffusers
 Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/diffusers/graphs/contributors)
@@ -23,81 +23,81 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: importlib-metadata
 Requires-Dist: filelock
-Requires-Dist: huggingface-hub (>=0.23.2)
+Requires-Dist: huggingface-hub>=0.23.2
 Requires-Dist: numpy
-Requires-Dist: regex (!=2019.12.17)
+Requires-Dist: regex!=2019.12.17
 Requires-Dist: requests
-Requires-Dist: safetensors (>=0.3.1)
+Requires-Dist: safetensors>=0.3.1
 Requires-Dist: Pillow
 Provides-Extra: dev
-Requires-Dist: urllib3 (<=2.0.0) ; extra == 'dev'
-Requires-Dist: isort (>=5.5.4) ; extra == 'dev'
-Requires-Dist: ruff (==0.1.5) ; extra == 'dev'
-Requires-Dist: hf-doc-builder (>=0.3.0) ; extra == 'dev'
-Requires-Dist: compel (==0.1.8) ; extra == 'dev'
-Requires-Dist: GitPython (<3.1.19) ; extra == 'dev'
-Requires-Dist: datasets ; extra == 'dev'
-Requires-Dist: Jinja2 ; extra == 'dev'
-Requires-Dist: invisible-watermark (>=0.2.0) ; extra == 'dev'
-Requires-Dist: k-diffusion (>=0.0.12) ; extra == 'dev'
-Requires-Dist: librosa ; extra == 'dev'
-Requires-Dist: parameterized ; extra == 'dev'
-Requires-Dist: pytest ; extra == 'dev'
-Requires-Dist: pytest-timeout ; extra == 'dev'
-Requires-Dist: pytest-xdist ; extra == 'dev'
-Requires-Dist: requests-mock (==1.10.0) ; extra == 'dev'
-Requires-Dist: safetensors (>=0.3.1) ; extra == 'dev'
-Requires-Dist: sentencepiece (!=0.1.92,>=0.1.91) ; extra == 'dev'
-Requires-Dist: scipy ; extra == 'dev'
-Requires-Dist: torchvision ; extra == 'dev'
-Requires-Dist: transformers (>=4.41.2) ; extra == 'dev'
-Requires-Dist: accelerate (>=0.31.0) ; extra == 'dev'
-Requires-Dist: protobuf (<4,>=3.20.3) ; extra == 'dev'
-Requires-Dist: tensorboard ; extra == 'dev'
-Requires-Dist: peft (>=0.6.0) ; extra == 'dev'
-Requires-Dist: torch (>=1.4) ; extra == 'dev'
-Requires-Dist: jax (>=0.4.1) ; extra == 'dev'
-Requires-Dist: jaxlib (>=0.4.1) ; extra == 'dev'
-Requires-Dist: flax (>=0.4.1) ; extra == 'dev'
+Requires-Dist: urllib3<=2.0.0; extra == "dev"
+Requires-Dist: isort>=5.5.4; extra == "dev"
+Requires-Dist: ruff==0.1.5; extra == "dev"
+Requires-Dist: hf-doc-builder>=0.3.0; extra == "dev"
+Requires-Dist: compel==0.1.8; extra == "dev"
+Requires-Dist: GitPython<3.1.19; extra == "dev"
+Requires-Dist: datasets; extra == "dev"
+Requires-Dist: Jinja2; extra == "dev"
+Requires-Dist: invisible-watermark>=0.2.0; extra == "dev"
+Requires-Dist: k-diffusion>=0.0.12; extra == "dev"
+Requires-Dist: librosa; extra == "dev"
+Requires-Dist: parameterized; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-timeout; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: requests-mock==1.10.0; extra == "dev"
+Requires-Dist: safetensors>=0.3.1; extra == "dev"
+Requires-Dist: sentencepiece!=0.1.92,>=0.1.91; extra == "dev"
+Requires-Dist: scipy; extra == "dev"
+Requires-Dist: torchvision; extra == "dev"
+Requires-Dist: transformers>=4.41.2; extra == "dev"
+Requires-Dist: accelerate>=0.31.0; extra == "dev"
+Requires-Dist: protobuf<4,>=3.20.3; extra == "dev"
+Requires-Dist: tensorboard; extra == "dev"
+Requires-Dist: peft>=0.6.0; extra == "dev"
+Requires-Dist: torch>=1.4; extra == "dev"
+Requires-Dist: jax>=0.4.1; extra == "dev"
+Requires-Dist: jaxlib>=0.4.1; extra == "dev"
+Requires-Dist: flax>=0.4.1; extra == "dev"
 Provides-Extra: docs
-Requires-Dist: hf-doc-builder (>=0.3.0) ; extra == 'docs'
+Requires-Dist: hf-doc-builder>=0.3.0; extra == "docs"
 Provides-Extra: flax
-Requires-Dist: jax (>=0.4.1) ; extra == 'flax'
-Requires-Dist: jaxlib (>=0.4.1) ; extra == 'flax'
-Requires-Dist: flax (>=0.4.1) ; extra == 'flax'
+Requires-Dist: jax>=0.4.1; extra == "flax"
+Requires-Dist: jaxlib>=0.4.1; extra == "flax"
+Requires-Dist: flax>=0.4.1; extra == "flax"
 Provides-Extra: quality
-Requires-Dist: urllib3 (<=2.0.0) ; extra == 'quality'
-Requires-Dist: isort (>=5.5.4) ; extra == 'quality'
-Requires-Dist: ruff (==0.1.5) ; extra == 'quality'
-Requires-Dist: hf-doc-builder (>=0.3.0) ; extra == 'quality'
+Requires-Dist: urllib3<=2.0.0; extra == "quality"
+Requires-Dist: isort>=5.5.4; extra == "quality"
+Requires-Dist: ruff==0.1.5; extra == "quality"
+Requires-Dist: hf-doc-builder>=0.3.0; extra == "quality"
 Provides-Extra: test
-Requires-Dist: compel (==0.1.8) ; extra == 'test'
-Requires-Dist: GitPython (<3.1.19) ; extra == 'test'
-Requires-Dist: datasets ; extra == 'test'
-Requires-Dist: Jinja2 ; extra == 'test'
-Requires-Dist: invisible-watermark (>=0.2.0) ; extra == 'test'
-Requires-Dist: k-diffusion (>=0.0.12) ; extra == 'test'
-Requires-Dist: librosa ; extra == 'test'
-Requires-Dist: parameterized ; extra == 'test'
-Requires-Dist: pytest ; extra == 'test'
-Requires-Dist: pytest-timeout ; extra == 'test'
-Requires-Dist: pytest-xdist ; extra == 'test'
-Requires-Dist: requests-mock (==1.10.0) ; extra == 'test'
-Requires-Dist: safetensors (>=0.3.1) ; extra == 'test'
-Requires-Dist: sentencepiece (!=0.1.92,>=0.1.91) ; extra == 'test'
-Requires-Dist: scipy ; extra == 'test'
-Requires-Dist: torchvision ; extra == 'test'
-Requires-Dist: transformers (>=4.41.2) ; extra == 'test'
+Requires-Dist: compel==0.1.8; extra == "test"
+Requires-Dist: GitPython<3.1.19; extra == "test"
+Requires-Dist: datasets; extra == "test"
+Requires-Dist: Jinja2; extra == "test"
+Requires-Dist: invisible-watermark>=0.2.0; extra == "test"
+Requires-Dist: k-diffusion>=0.0.12; extra == "test"
+Requires-Dist: librosa; extra == "test"
+Requires-Dist: parameterized; extra == "test"
+Requires-Dist: pytest; extra == "test"
+Requires-Dist: pytest-timeout; extra == "test"
+Requires-Dist: pytest-xdist; extra == "test"
+Requires-Dist: requests-mock==1.10.0; extra == "test"
+Requires-Dist: safetensors>=0.3.1; extra == "test"
+Requires-Dist: sentencepiece!=0.1.92,>=0.1.91; extra == "test"
+Requires-Dist: scipy; extra == "test"
+Requires-Dist: torchvision; extra == "test"
+Requires-Dist: transformers>=4.41.2; extra == "test"
 Provides-Extra: torch
-Requires-Dist: torch (>=1.4) ; extra == 'torch'
-Requires-Dist: accelerate (>=0.31.0) ; extra == 'torch'
+Requires-Dist: torch>=1.4; extra == "torch"
+Requires-Dist: accelerate>=0.31.0; extra == "torch"
 Provides-Extra: training
-Requires-Dist: accelerate (>=0.31.0) ; extra == 'training'
-Requires-Dist: datasets ; extra == 'training'
-Requires-Dist: protobuf (<4,>=3.20.3) ; extra == 'training'
-Requires-Dist: tensorboard ; extra == 'training'
-Requires-Dist: Jinja2 ; extra == 'training'
-Requires-Dist: peft (>=0.6.0) ; extra == 'training'
+Requires-Dist: accelerate>=0.31.0; extra == "training"
+Requires-Dist: datasets; extra == "training"
+Requires-Dist: protobuf<4,>=3.20.3; extra == "training"
+Requires-Dist: tensorboard; extra == "training"
+Requires-Dist: Jinja2; extra == "training"
+Requires-Dist: peft>=0.6.0; extra == "training"
 
 <!---
 Copyright 2022 - The HuggingFace Team. All rights reserved.

{diffusers-0.32.0.dist-info → diffusers-0.32.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-diffusers/__init__.py,sha256=1Wj313G6Z6unQw62O1Dtmo0w9edqui1CREmYSHdHMCM,38255
+diffusers/__init__.py,sha256=3K4OWRY9hMeZ-iKFBRnqBKqt7pwirK0ahM3L_AbPzyM,38255
 diffusers/callbacks.py,sha256=wmLFSUrnEZ9gz6gay2pRWfha5cErTSAbB9PK482_KGY,8749
 diffusers/configuration_utils.py,sha256=FQ5zC2BwAhIv1S5NcP02_kJjFlhDjo6F76snd1n6RXQ,33054
 diffusers/dependency_versions_check.py,sha256=J_ZAEhVN6uLWAOUZCJrcGJ7PYxUek4f_nwGTFM7LTk8,1271
@@ -49,7 +49,7 @@ diffusers/models/modeling_flax_pytorch_utils.py,sha256=h8KonTFgb_-4RnESXhJGeuW_b
 diffusers/models/modeling_flax_utils.py,sha256=84Cj8NqhPnIrqk6c6ujPfzqiJFI4XQovKKkCW_xvDI0,26954
 diffusers/models/modeling_outputs.py,sha256=XH3sJO34MRW6UuWqqKo05mVqxGSBFRazpap_-YLwO2I,1042
 diffusers/models/modeling_pytorch_flax_utils.py,sha256=sEf_jVR2nF0_derGLAOKIfSUc7HWNLM61RTXDLGoE7A,6973
-diffusers/models/modeling_utils.py,sha256=ASXSmlzsPRuPXIiApv4jxK3VP-nyrJ0Kjtq3ipCz6-A,70745
+diffusers/models/modeling_utils.py,sha256=JiE60IrHcHJSlxbcStup3oOjR3QxuMhIOHEqxDxda7I,70626
 diffusers/models/normalization.py,sha256=MR1Xry19rMNW-1WaeIUd8ht5kfORu_yJwATYhgnjsbA,22238
 diffusers/models/resnet.py,sha256=ML9EdypGYniSay_EsyswuTlmGi7429WJhYqIW7VEBoQ,32241
 diffusers/models/resnet_flax.py,sha256=tqRZQCZIq7NlXex3eGldyhRpZjr_EXWl1l2eVflFV7c,4021
@@ -458,7 +458,7 @@ diffusers/quantizers/gguf/__init__.py,sha256=2bxvfZbFr4xqm953cZaGJMgSCRiGJAWwbwK
 diffusers/quantizers/gguf/gguf_quantizer.py,sha256=sYJUfFhBOYjsM1dB_a-h5S-IFmu5nEh8cmKMnibcIyk,5754
 diffusers/quantizers/gguf/utils.py,sha256=3QMkB74wr01OwBeaqv9WC4Fw4VwV7xeOPYOso1MvfbI,15903
 diffusers/quantizers/torchao/__init__.py,sha256=A82P7eWAKFVZ8MlWh1seo1SmF93wsxXXd6i8dnMlALI,662
-diffusers/quantizers/torchao/torchao_quantizer.py,sha256=yVBtQJGWd6zCnlCCaGhZ7Rk54mSgf0uZb2hb3jSClg0,12512
+diffusers/quantizers/torchao/torchao_quantizer.py,sha256=6Ktpd6ZXqXkOLWtqoP3XmEOvwDeV_2_MhviGu6eTLPA,12821
 diffusers/schedulers/__init__.py,sha256=56Cgo7azvjQEMkUqVf2-psEJ8MoGaZDQ8-LtGV69brg,10927
 diffusers/schedulers/scheduling_amused.py,sha256=pioDeoYfXWP2CnAiI3-lczDdfSbkfXRC9m9REN_kmvI,6590
 diffusers/schedulers/scheduling_consistency_decoder.py,sha256=IKNAkeVZIxTPk_hS6QBvFbkjv_h8yWbtU3FmIUSXKVI,6817
@@ -542,9 +542,9 @@ diffusers/utils/state_dict_utils.py,sha256=NsWzyX4eqKCfjLjgChQnFSf7nSQz1XFHgINYB
 diffusers/utils/testing_utils.py,sha256=zIf3SG35pr2krYFyODxASzkYyAuHZxWNQcxTCmVHCV0,41267
 diffusers/utils/torch_utils.py,sha256=p9VYa8VH4j_QgbIWHxfuzfhvdMKRn-gmQI1-BjPdGBY,6345
 diffusers/utils/versions.py,sha256=-e7XW1TzZ-tsRo9PMQHp-hNGYHuVDFzLtwg3uAJzqdI,4333
-diffusers-0.32.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-diffusers-0.32.0.dist-info/METADATA,sha256=zgpdWIbC12p0kwcmzHTPnio22EIJftgKAA0hGb1oRPg,19140
-diffusers-0.32.0.dist-info/WHEEL,sha256=2wepM1nk4DS4eFpYrW1TTqPcoGNfHhhO_i5m4cOimbo,92
-diffusers-0.32.0.dist-info/entry_points.txt,sha256=_1bvshKV_6_b63_FAkcUs9W6tUKGeIoQ3SHEZsovEWs,72
-diffusers-0.32.0.dist-info/top_level.txt,sha256=axJl2884vMSvhzrFrSoht36QXA_6gZN9cKtg4xOO72o,10
-diffusers-0.32.0.dist-info/RECORD,,
+diffusers-0.32.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+diffusers-0.32.1.dist-info/METADATA,sha256=gtZLuGgEmIFA9EH8ARNTGK0lA9UuKC0u_GFGSXRsL3w,18949
+diffusers-0.32.1.dist-info/WHEEL,sha256=P9jw-gEje8ByB7_hXoICnHtVCrEwMQh-630tKvQWehc,91
+diffusers-0.32.1.dist-info/entry_points.txt,sha256=_1bvshKV_6_b63_FAkcUs9W6tUKGeIoQ3SHEZsovEWs,72
+diffusers-0.32.1.dist-info/top_level.txt,sha256=axJl2884vMSvhzrFrSoht36QXA_6gZN9cKtg4xOO72o,10
+diffusers-0.32.1.dist-info/RECORD,,

{diffusers-0.32.0.dist-info → diffusers-0.32.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.38.4)
+Generator: setuptools (75.3.0)
 Root-Is-Purelib: true
 Tag: py3-none-any