diffusers 0.24.0__py3-none-any.whl → 0.25.0__py3-none-any.whl

diffusers/utils/hub_utils.py

@@ -25,20 +25,21 @@ from typing import Dict, Optional, Union
 from uuid import uuid4
 
 from huggingface_hub import (
-    HfFolder,
     ModelCard,
     ModelCardData,
     create_repo,
+    get_full_repo_name,
     hf_hub_download,
     upload_folder,
-    whoami,
 )
+from huggingface_hub.constants import HF_HUB_CACHE, HF_HUB_DISABLE_TELEMETRY, HF_HUB_OFFLINE
 from huggingface_hub.file_download import REGEX_COMMIT_HASH
 from huggingface_hub.utils import (
     EntryNotFoundError,
     RepositoryNotFoundError,
     RevisionNotFoundError,
     is_jinja_available,
+    validate_hf_hub_args,
 )
 from packaging import version
 from requests import HTTPError
@@ -46,7 +47,6 @@ from requests import HTTPError
 from .. import __version__
 from .constants import (
     DEPRECATED_REVISION_ARGS,
-    DIFFUSERS_CACHE,
     HUGGINGFACE_CO_RESOLVE_ENDPOINT,
     SAFETENSORS_WEIGHTS_NAME,
     WEIGHTS_NAME,
@@ -69,9 +69,6 @@ logger = get_logger(__name__)
 
 MODEL_CARD_TEMPLATE_PATH = Path(__file__).parent / "model_card_template.md"
 SESSION_ID = uuid4().hex
-HF_HUB_OFFLINE = os.getenv("HF_HUB_OFFLINE", "").upper() in ENV_VARS_TRUE_VALUES
-DISABLE_TELEMETRY = os.getenv("DISABLE_TELEMETRY", "").upper() in ENV_VARS_TRUE_VALUES
-HUGGINGFACE_CO_TELEMETRY = HUGGINGFACE_CO_RESOLVE_ENDPOINT + "/api/telemetry/"
 
 
 def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str:
@@ -79,7 +76,7 @@ def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str:
     Formats a user-agent string with basic info about a request.
     """
     ua = f"diffusers/{__version__}; python/{sys.version.split()[0]}; session_id/{SESSION_ID}"
-    if DISABLE_TELEMETRY or HF_HUB_OFFLINE:
+    if HF_HUB_DISABLE_TELEMETRY or HF_HUB_OFFLINE:
         return ua + "; telemetry/off"
     if is_torch_available():
         ua += f"; torch/{_torch_version}"
@@ -98,16 +95,6 @@ def http_user_agent(user_agent: Union[Dict, str, None] = None) -> str:
     return ua
 
 
-def get_full_repo_name(model_id: str, organization: Optional[str] = None, token: Optional[str] = None):
-    if token is None:
-        token = HfFolder.get_token()
-    if organization is None:
-        username = whoami(token)["name"]
-        return f"{username}/{model_id}"
-    else:
-        return f"{organization}/{model_id}"
-
-
 def create_model_card(args, model_name):
     if not is_jinja_available():
         raise ValueError(
@@ -183,7 +170,7 @@ old_diffusers_cache = os.path.join(hf_cache_home, "diffusers")
 
 def move_cache(old_cache_dir: Optional[str] = None, new_cache_dir: Optional[str] = None) -> None:
     if new_cache_dir is None:
-        new_cache_dir = DIFFUSERS_CACHE
+        new_cache_dir = HF_HUB_CACHE
     if old_cache_dir is None:
         old_cache_dir = old_diffusers_cache
 
@@ -203,7 +190,7 @@ def move_cache(old_cache_dir: Optional[str] = None, new_cache_dir: Optional[str]
     # At this point, old_cache_dir contains symlinks to the new cache (it can still be used).
 
 
-cache_version_file = os.path.join(DIFFUSERS_CACHE, "version_diffusers_cache.txt")
+cache_version_file = os.path.join(HF_HUB_CACHE, "version_diffusers_cache.txt")
 if not os.path.isfile(cache_version_file):
     cache_version = 0
 else:
@@ -233,12 +220,12 @@ if cache_version < 1:
 
 if cache_version < 1:
     try:
-        os.makedirs(DIFFUSERS_CACHE, exist_ok=True)
+        os.makedirs(HF_HUB_CACHE, exist_ok=True)
         with open(cache_version_file, "w") as f:
             f.write("1")
     except Exception:
         logger.warning(
-            f"There was a problem when trying to write in your cache folder ({DIFFUSERS_CACHE}). Please, ensure "
+            f"There was a problem when trying to write in your cache folder ({HF_HUB_CACHE}). Please, ensure "
             "the directory exists and can be written to."
         )
 
@@ -252,20 +239,21 @@ def _add_variant(weights_name: str, variant: Optional[str] = None) -> str:
     return weights_name
 
 
+@validate_hf_hub_args
 def _get_model_file(
-    pretrained_model_name_or_path,
+    pretrained_model_name_or_path: Union[str, Path],
     *,
-    weights_name,
-    subfolder,
-    cache_dir,
-    force_download,
-    proxies,
-    resume_download,
-    local_files_only,
-    use_auth_token,
-    user_agent,
-    revision,
-    commit_hash=None,
+    weights_name: str,
+    subfolder: Optional[str],
+    cache_dir: Optional[str],
+    force_download: bool,
+    proxies: Optional[Dict],
+    resume_download: bool,
+    local_files_only: bool,
+    token: Optional[str],
+    user_agent: Union[Dict, str, None],
+    revision: Optional[str],
+    commit_hash: Optional[str] = None,
 ):
     pretrained_model_name_or_path = str(pretrained_model_name_or_path)
     if os.path.isfile(pretrained_model_name_or_path):
@@ -300,7 +288,7 @@ def _get_model_file(
                     proxies=proxies,
                     resume_download=resume_download,
                     local_files_only=local_files_only,
-                    use_auth_token=use_auth_token,
+                    token=token,
                     user_agent=user_agent,
                     subfolder=subfolder,
                     revision=revision or commit_hash,
@@ -325,7 +313,7 @@ def _get_model_file(
                 proxies=proxies,
                 resume_download=resume_download,
                 local_files_only=local_files_only,
-                use_auth_token=use_auth_token,
+                token=token,
                 user_agent=user_agent,
                 subfolder=subfolder,
                 revision=revision or commit_hash,
@@ -336,7 +324,7 @@ def _get_model_file(
             raise EnvironmentError(
                 f"{pretrained_model_name_or_path} is not a local folder and is not a valid model identifier "
                 "listed on 'https://huggingface.co/models'\nIf this is a private repository, make sure to pass a "
-                "token having permission to this repo with `use_auth_token` or log in with `huggingface-cli "
+                "token having permission to this repo with `token` or log in with `huggingface-cli "
                 "login`."
             )
         except RevisionNotFoundError:

diffusers/utils/logging.py

@@ -213,7 +213,7 @@ def remove_handler(handler: logging.Handler) -> None:
 
     _configure_library_root_logger()
 
-    assert handler is not None and handler not in _get_library_root_logger().handlers
+    assert handler is not None and handler in _get_library_root_logger().handlers
     _get_library_root_logger().removeHandler(handler)
 
 

diffusers/utils/state_dict_utils.py

@@ -79,6 +79,14 @@ PEFT_TO_DIFFUSERS = {
     ".v_proj.lora_A": ".v_proj.lora_linear_layer.down",
     ".out_proj.lora_B": ".out_proj.lora_linear_layer.up",
     ".out_proj.lora_A": ".out_proj.lora_linear_layer.down",
+    "to_k.lora_A": "to_k.lora.down",
+    "to_k.lora_B": "to_k.lora.up",
+    "to_q.lora_A": "to_q.lora.down",
+    "to_q.lora_B": "to_q.lora.up",
+    "to_v.lora_A": "to_v.lora.down",
+    "to_v.lora_B": "to_v.lora.up",
+    "to_out.0.lora_A": "to_out.0.lora.down",
+    "to_out.0.lora_B": "to_out.0.lora.up",
 }
 
 DIFFUSERS_OLD_TO_DIFFUSERS = {

diffusers/utils/testing_utils.py

@@ -17,7 +17,7 @@ from contextlib import contextmanager
 from distutils.util import strtobool
 from io import BytesIO, StringIO
 from pathlib import Path
-from typing import List, Optional, Union
+from typing import Callable, Dict, List, Optional, Union
 
 import numpy as np
 import PIL.Image
@@ -58,6 +58,17 @@ USE_PEFT_BACKEND = _required_peft_version and _required_transformers_version
 if is_torch_available():
     import torch
 
+    # Set a backend environment variable for any extra module import required for a custom accelerator
+    if "DIFFUSERS_TEST_BACKEND" in os.environ:
+        backend = os.environ["DIFFUSERS_TEST_BACKEND"]
+        try:
+            _ = importlib.import_module(backend)
+        except ModuleNotFoundError as e:
+            raise ModuleNotFoundError(
+                f"Failed to import `DIFFUSERS_TEST_BACKEND` '{backend}'! This should be the name of an installed module \
+                    to enable a specified backend.):\n{e}"
+            ) from e
+
     if "DIFFUSERS_TEST_DEVICE" in os.environ:
         torch_device = os.environ["DIFFUSERS_TEST_DEVICE"]
         try:
@@ -210,6 +221,36 @@ def require_torch_gpu(test_case):
     )
 
 
+# These decorators are for accelerator-specific behaviours that are not GPU-specific
+def require_torch_accelerator(test_case):
+    """Decorator marking a test that requires an accelerator backend and PyTorch."""
+    return unittest.skipUnless(is_torch_available() and torch_device != "cpu", "test requires accelerator+PyTorch")(
+        test_case
+    )
+
+
+def require_torch_accelerator_with_fp16(test_case):
+    """Decorator marking a test that requires an accelerator with support for the FP16 data type."""
+    return unittest.skipUnless(_is_torch_fp16_available(torch_device), "test requires accelerator with fp16 support")(
+        test_case
+    )
+
+
+def require_torch_accelerator_with_fp64(test_case):
+    """Decorator marking a test that requires an accelerator with support for the FP64 data type."""
+    return unittest.skipUnless(_is_torch_fp64_available(torch_device), "test requires accelerator with fp64 support")(
+        test_case
+    )
+
+
+def require_torch_accelerator_with_training(test_case):
+    """Decorator marking a test that requires an accelerator with support for training."""
+    return unittest.skipUnless(
+        is_torch_available() and backend_supports_training(torch_device),
+        "test requires accelerator with training support",
+    )(test_case)
+
+
 def skip_mps(test_case):
     """Decorator marking a test to skip if torch_device is 'mps'"""
     return unittest.skipUnless(torch_device != "mps", "test requires non 'mps' device")(test_case)
@@ -259,6 +300,23 @@ def require_peft_backend(test_case):
     return unittest.skipUnless(USE_PEFT_BACKEND, "test requires PEFT backend")(test_case)
 
 
+def require_peft_version_greater(peft_version):
+    """
+    Decorator marking a test that requires PEFT backend with a specific version, this would require some specific
+    versions of PEFT and transformers.
+    """
+
+    def decorator(test_case):
+        correct_peft_version = is_peft_available() and version.parse(
+            version.parse(importlib.metadata.version("peft")).base_version
+        ) > version.parse(peft_version)
+        return unittest.skipUnless(
+            correct_peft_version, f"test requires PEFT backend with the version greater than {peft_version}"
+        )(test_case)
+
+    return decorator
+
+
 def deprecate_after_peft_backend(test_case):
     """
     Decorator marking a test that will be skipped after PEFT backend
@@ -766,3 +824,143 @@ def disable_full_determinism():
     os.environ["CUDA_LAUNCH_BLOCKING"] = "0"
     os.environ["CUBLAS_WORKSPACE_CONFIG"] = ""
     torch.use_deterministic_algorithms(False)
+
+
+# Utils for custom and alternative accelerator devices
+def _is_torch_fp16_available(device):
+    if not is_torch_available():
+        return False
+
+    import torch
+
+    device = torch.device(device)
+
+    try:
+        x = torch.zeros((2, 2), dtype=torch.float16).to(device)
+        _ = torch.mul(x, x)
+        return True
+
+    except Exception as e:
+        if device.type == "cuda":
+            raise ValueError(
+                f"You have passed a device of type 'cuda' which should work with 'fp16', but 'cuda' does not seem to be correctly installed on your machine: {e}"
+            )
+
+        return False
+
+
+def _is_torch_fp64_available(device):
+    if not is_torch_available():
+        return False
+
+    import torch
+
+    try:
+        x = torch.zeros((2, 2), dtype=torch.float64).to(device)
+        _ = torch.mul(x, x)
+        return True
+
+    except Exception as e:
+        if device.type == "cuda":
+            raise ValueError(
+                f"You have passed a device of type 'cuda' which should work with 'fp64', but 'cuda' does not seem to be correctly installed on your machine: {e}"
+            )
+
+        return False
+
+
+# Guard these lookups for when Torch is not used - alternative accelerator support is for PyTorch
+if is_torch_available():
+    # Behaviour flags
+    BACKEND_SUPPORTS_TRAINING = {"cuda": True, "cpu": True, "mps": False, "default": True}
+
+    # Function definitions
+    BACKEND_EMPTY_CACHE = {"cuda": torch.cuda.empty_cache, "cpu": None, "mps": None, "default": None}
+    BACKEND_DEVICE_COUNT = {"cuda": torch.cuda.device_count, "cpu": lambda: 0, "mps": lambda: 0, "default": 0}
+    BACKEND_MANUAL_SEED = {"cuda": torch.cuda.manual_seed, "cpu": torch.manual_seed, "default": torch.manual_seed}
+
+
+# This dispatches a defined function according to the accelerator from the function definitions.
+def _device_agnostic_dispatch(device: str, dispatch_table: Dict[str, Callable], *args, **kwargs):
+    if device not in dispatch_table:
+        return dispatch_table["default"](*args, **kwargs)
+
+    fn = dispatch_table[device]
+
+    # Some device agnostic functions return values. Need to guard against 'None' instead at
+    # user level
+    if fn is None:
+        return None
+
+    return fn(*args, **kwargs)
+
+
+# These are callables which automatically dispatch the function specific to the accelerator
+def backend_manual_seed(device: str, seed: int):
+    return _device_agnostic_dispatch(device, BACKEND_MANUAL_SEED, seed)
+
+
+def backend_empty_cache(device: str):
+    return _device_agnostic_dispatch(device, BACKEND_EMPTY_CACHE)
+
+
+def backend_device_count(device: str):
+    return _device_agnostic_dispatch(device, BACKEND_DEVICE_COUNT)
+
+
+# These are callables which return boolean behaviour flags and can be used to specify some
+# device agnostic alternative where the feature is unsupported.
+def backend_supports_training(device: str):
+    if not is_torch_available():
+        return False
+
+    if device not in BACKEND_SUPPORTS_TRAINING:
+        device = "default"
+
+    return BACKEND_SUPPORTS_TRAINING[device]
+
+
+# Guard for when Torch is not available
+if is_torch_available():
+    # Update device function dict mapping
+    def update_mapping_from_spec(device_fn_dict: Dict[str, Callable], attribute_name: str):
+        try:
+            # Try to import the function directly
+            spec_fn = getattr(device_spec_module, attribute_name)
+            device_fn_dict[torch_device] = spec_fn
+        except AttributeError as e:
+            # If the function doesn't exist, and there is no default, throw an error
+            if "default" not in device_fn_dict:
+                raise AttributeError(
+                    f"`{attribute_name}` not found in '{device_spec_path}' and no default fallback function found."
+                ) from e
+
+    if "DIFFUSERS_TEST_DEVICE_SPEC" in os.environ:
+        device_spec_path = os.environ["DIFFUSERS_TEST_DEVICE_SPEC"]
+        if not Path(device_spec_path).is_file():
+            raise ValueError(f"Specified path to device specification file is not found. Received {device_spec_path}")
+
+        try:
+            import_name = device_spec_path[: device_spec_path.index(".py")]
+        except ValueError as e:
+            raise ValueError(f"Provided device spec file is not a Python file! Received {device_spec_path}") from e
+
+        device_spec_module = importlib.import_module(import_name)
+
+        try:
+            device_name = device_spec_module.DEVICE_NAME
+        except AttributeError:
+            raise AttributeError("Device spec file did not contain `DEVICE_NAME`")
+
+        if "DIFFUSERS_TEST_DEVICE" in os.environ and torch_device != device_name:
+            msg = f"Mismatch between environment variable `DIFFUSERS_TEST_DEVICE` '{torch_device}' and device found in spec '{device_name}'\n"
+            msg += "Either unset `DIFFUSERS_TEST_DEVICE` or ensure it matches device spec name."
+            raise ValueError(msg)
+
+        torch_device = device_name
+
+        # Add one entry here for each `BACKEND_*` dictionary.
+        update_mapping_from_spec(BACKEND_MANUAL_SEED, "MANUAL_SEED_FN")
+        update_mapping_from_spec(BACKEND_EMPTY_CACHE, "EMPTY_CACHE_FN")
+        update_mapping_from_spec(BACKEND_DEVICE_COUNT, "DEVICE_COUNT_FN")
+        update_mapping_from_spec(BACKEND_SUPPORTS_TRAINING, "SUPPORTS_TRAINING")

diffusers/utils/torch_utils.py

@@ -89,7 +89,7 @@ def is_compiled_module(module) -> bool:
     return isinstance(module, torch._dynamo.eval_frame.OptimizedModule)
 
 
-def fourier_filter(x_in: torch.Tensor, threshold: int, scale: int) -> torch.Tensor:
+def fourier_filter(x_in: "torch.Tensor", threshold: int, scale: int) -> "torch.Tensor":
     """Fourier filter as introduced in FreeU (https://arxiv.org/abs/2309.11497).
 
     This version of the method comes from here:
@@ -121,8 +121,8 @@ def fourier_filter(x_in: torch.Tensor, threshold: int, scale: int) -> torch.Tens
 
 
 def apply_freeu(
-    resolution_idx: int, hidden_states: torch.Tensor, res_hidden_states: torch.Tensor, **freeu_kwargs
-) -> Tuple[torch.Tensor, torch.Tensor]:
+    resolution_idx: int, hidden_states: "torch.Tensor", res_hidden_states: "torch.Tensor", **freeu_kwargs
+) -> Tuple["torch.Tensor", "torch.Tensor"]:
     """Applies the FreeU mechanism as introduced in https:
     //arxiv.org/abs/2309.11497. Adapted from the official code repository: https://github.com/ChenyangSi/FreeU.
 

{diffusers-0.24.0.dist-info → diffusers-0.25.0.dist-info}/METADATA

@@ -1,13 +1,12 @@
 Metadata-Version: 2.1
 Name: diffusers
-Version: 0.24.0
+Version: 0.25.0
 Summary: State-of-the-art diffusion in PyTorch and JAX.
 Home-page: https://github.com/huggingface/diffusers
-Author: The HuggingFace team
+Author: The Hugging Face team (past and future) with the help of all our contributors (https://github.com/huggingface/diffusers/graphs/contributors)
 Author-email: patrick@huggingface.co
-License: Apache
+License: Apache 2.0 License
 Keywords: deep learning diffusion jax pytorch stable diffusion audioldm
-Platform: UNKNOWN
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Education
@@ -22,81 +21,85 @@ Classifier: Programming Language :: Python :: 3.10
 Requires-Python: >=3.8.0
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: Pillow
-Requires-Dist: filelock
-Requires-Dist: huggingface-hub (>=0.19.4)
 Requires-Dist: importlib-metadata
+Requires-Dist: filelock
+Requires-Dist: huggingface-hub >=0.19.4
 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: Jinja2 ; extra == 'dev'
-Requires-Dist: accelerate (>=0.11.0) ; extra == 'dev'
-Requires-Dist: compel (==0.1.8) ; 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: flax (>=0.4.1) ; extra == 'dev'
-Requires-Dist: hf-doc-builder (>=0.3.0) ; extra == 'dev'
-Requires-Dist: invisible-watermark (>=0.2.0) ; extra == 'dev'
-Requires-Dist: isort (>=5.5.4) ; extra == 'dev'
-Requires-Dist: jax (>=0.4.1) ; extra == 'dev'
-Requires-Dist: jaxlib (>=0.4.1) ; extra == 'dev'
-Requires-Dist: k-diffusion (>=0.0.12) ; 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: omegaconf ; extra == 'dev'
 Requires-Dist: parameterized ; extra == 'dev'
-Requires-Dist: protobuf (<4,>=3.20.3) ; 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: ruff (<=0.2,>=0.1.5) ; extra == 'dev'
-Requires-Dist: safetensors (>=0.3.1) ; 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: sentencepiece (!=0.1.92,>=0.1.91) ; extra == 'dev'
-Requires-Dist: tensorboard ; extra == 'dev'
-Requires-Dist: torch (>=1.4) ; extra == 'dev'
 Requires-Dist: torchvision ; extra == 'dev'
-Requires-Dist: transformers (>=4.25.1) ; extra == 'dev'
-Requires-Dist: urllib3 (<=2.0.0) ; extra == 'dev'
+Requires-Dist: transformers >=4.25.1 ; extra == 'dev'
+Requires-Dist: accelerate >=0.11.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: flax (>=0.4.1) ; extra == 'flax'
-Requires-Dist: jax (>=0.4.1) ; extra == 'flax'
-Requires-Dist: jaxlib (>=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: hf-doc-builder (>=0.3.0) ; extra == 'quality'
-Requires-Dist: isort (>=5.5.4) ; extra == 'quality'
-Requires-Dist: ruff (<=0.2,>=0.1.5) ; extra == 'quality'
-Requires-Dist: urllib3 (<=2.0.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: Jinja2 ; extra == 'test'
-Requires-Dist: compel (==0.1.8) ; 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: invisible-watermark (>=0.2.0) ; extra == 'test'
-Requires-Dist: k-diffusion (>=0.0.12) ; 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: omegaconf ; 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: 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: sentencepiece (!=0.1.92,>=0.1.91) ; extra == 'test'
 Requires-Dist: torchvision ; extra == 'test'
-Requires-Dist: transformers (>=4.25.1) ; extra == 'test'
+Requires-Dist: transformers >=4.25.1 ; extra == 'test'
 Provides-Extra: torch
-Requires-Dist: accelerate (>=0.11.0) ; extra == 'torch'
-Requires-Dist: torch (>=1.4) ; extra == 'torch'
+Requires-Dist: torch >=1.4 ; extra == 'torch'
+Requires-Dist: accelerate >=0.11.0 ; extra == 'torch'
 Provides-Extra: training
-Requires-Dist: Jinja2 ; extra == 'training'
-Requires-Dist: accelerate (>=0.11.0) ; extra == 'training'
+Requires-Dist: accelerate >=0.11.0 ; extra == 'training'
 Requires-Dist: datasets ; extra == 'training'
-Requires-Dist: protobuf (<4,>=3.20.3) ; 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.
@@ -177,7 +180,7 @@ Please refer to the [How to use Stable Diffusion in Apple Silicon](https://huggi
 
 ## Quickstart
 
-Generating outputs is super easy with 🤗 Diffusers. To generate an image from text, use the `from_pretrained` method to load any pretrained diffusion model (browse the [Hub](https://huggingface.co/models?library=diffusers&sort=downloads) for 15000+ checkpoints):
+Generating outputs is super easy with 🤗 Diffusers. To generate an image from text, use the `from_pretrained` method to load any pretrained diffusion model (browse the [Hub](https://huggingface.co/models?library=diffusers&sort=downloads) for 16000+ checkpoints):
 
 ```python
 from diffusers import DiffusionPipeline
@@ -319,7 +322,7 @@ Also, say 👋 in our public Discord channel <a href="https://discord.gg/G7tWnz9
 - https://github.com/deep-floyd/IF
 - https://github.com/bentoml/BentoML
 - https://github.com/bmaltais/kohya_ss
-- +6000 other amazing GitHub repositories 💪
+- +7000 other amazing GitHub repositories 💪
 
 Thank you for using us ❤️.
 
@@ -346,5 +349,3 @@ We also want to thank @heejkoo for the very helpful overview of papers, code and
   howpublished = {\url{https://github.com/huggingface/diffusers}}
 }
 ```
-
-