diffusers 0.15.1__py3-none-any.whl → 0.16.1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- diffusers/__init__.py +7 -2
- diffusers/configuration_utils.py +4 -0
- diffusers/loaders.py +262 -12
- diffusers/models/attention.py +31 -12
- diffusers/models/attention_processor.py +189 -0
- diffusers/models/controlnet.py +9 -2
- diffusers/models/embeddings.py +66 -0
- diffusers/models/modeling_pytorch_flax_utils.py +6 -0
- diffusers/models/modeling_utils.py +5 -2
- diffusers/models/transformer_2d.py +1 -1
- diffusers/models/unet_2d_condition.py +45 -6
- diffusers/models/vae.py +3 -0
- diffusers/pipelines/__init__.py +8 -0
- diffusers/pipelines/alt_diffusion/modeling_roberta_series.py +25 -10
- diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion.py +8 -0
- diffusers/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py +8 -0
- diffusers/pipelines/audioldm/pipeline_audioldm.py +1 -1
- diffusers/pipelines/deepfloyd_if/__init__.py +54 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if.py +854 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img.py +979 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if_img2img_superresolution.py +1097 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting.py +1098 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if_inpainting_superresolution.py +1208 -0
- diffusers/pipelines/deepfloyd_if/pipeline_if_superresolution.py +947 -0
- diffusers/pipelines/deepfloyd_if/safety_checker.py +59 -0
- diffusers/pipelines/deepfloyd_if/timesteps.py +579 -0
- diffusers/pipelines/deepfloyd_if/watermark.py +46 -0
- diffusers/pipelines/pipeline_utils.py +54 -25
- diffusers/pipelines/stable_diffusion/convert_from_ckpt.py +37 -20
- diffusers/pipelines/stable_diffusion/pipeline_flax_stable_diffusion_controlnet.py +1 -1
- diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_upscale.py +12 -1
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py +10 -2
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py +10 -8
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py +59 -4
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py +9 -2
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py +10 -2
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +9 -2
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py +22 -12
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py +9 -2
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_pix2pix_zero.py +34 -30
- diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py +93 -10
- diffusers/pipelines/versatile_diffusion/modeling_text_unet.py +45 -6
- diffusers/schedulers/scheduling_ddpm.py +63 -16
- diffusers/schedulers/scheduling_heun_discrete.py +51 -1
- diffusers/utils/__init__.py +4 -1
- diffusers/utils/dummy_torch_and_transformers_objects.py +80 -5
- diffusers/utils/dynamic_modules_utils.py +1 -1
- diffusers/utils/hub_utils.py +4 -1
- diffusers/utils/import_utils.py +41 -0
- diffusers/utils/pil_utils.py +24 -0
- diffusers/utils/testing_utils.py +10 -0
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/METADATA +1 -1
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/RECORD +57 -47
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/LICENSE +0 -0
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/WHEEL +0 -0
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/entry_points.txt +0 -0
- {diffusers-0.15.1.dist-info → diffusers-0.16.1.dist-info}/top_level.txt +0 -0
@@ -0,0 +1,54 @@
|
|
1
|
+
from dataclasses import dataclass
|
2
|
+
from typing import List, Optional, Union
|
3
|
+
|
4
|
+
import numpy as np
|
5
|
+
import PIL
|
6
|
+
|
7
|
+
from ...utils import BaseOutput, OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
|
8
|
+
from .timesteps import (
|
9
|
+
fast27_timesteps,
|
10
|
+
smart27_timesteps,
|
11
|
+
smart50_timesteps,
|
12
|
+
smart100_timesteps,
|
13
|
+
smart185_timesteps,
|
14
|
+
super27_timesteps,
|
15
|
+
super40_timesteps,
|
16
|
+
super100_timesteps,
|
17
|
+
)
|
18
|
+
|
19
|
+
|
20
|
+
@dataclass
|
21
|
+
class IFPipelineOutput(BaseOutput):
|
22
|
+
"""
|
23
|
+
Args:
|
24
|
+
Output class for Stable Diffusion pipelines.
|
25
|
+
images (`List[PIL.Image.Image]` or `np.ndarray`)
|
26
|
+
List of denoised PIL images of length `batch_size` or numpy array of shape `(batch_size, height, width,
|
27
|
+
num_channels)`. PIL images or numpy array present the denoised images of the diffusion pipeline.
|
28
|
+
nsfw_detected (`List[bool]`)
|
29
|
+
List of flags denoting whether the corresponding generated image likely represents "not-safe-for-work"
|
30
|
+
(nsfw) content or a watermark. `None` if safety checking could not be performed.
|
31
|
+
watermark_detected (`List[bool]`)
|
32
|
+
List of flags denoting whether the corresponding generated image likely has a watermark. `None` if safety
|
33
|
+
checking could not be performed.
|
34
|
+
"""
|
35
|
+
|
36
|
+
images: Union[List[PIL.Image.Image], np.ndarray]
|
37
|
+
nsfw_detected: Optional[List[bool]]
|
38
|
+
watermark_detected: Optional[List[bool]]
|
39
|
+
|
40
|
+
|
41
|
+
try:
|
42
|
+
if not (is_transformers_available() and is_torch_available()):
|
43
|
+
raise OptionalDependencyNotAvailable()
|
44
|
+
except OptionalDependencyNotAvailable:
|
45
|
+
from ...utils.dummy_torch_and_transformers_objects import * # noqa F403
|
46
|
+
else:
|
47
|
+
from .pipeline_if import IFPipeline
|
48
|
+
from .pipeline_if_img2img import IFImg2ImgPipeline
|
49
|
+
from .pipeline_if_img2img_superresolution import IFImg2ImgSuperResolutionPipeline
|
50
|
+
from .pipeline_if_inpainting import IFInpaintingPipeline
|
51
|
+
from .pipeline_if_inpainting_superresolution import IFInpaintingSuperResolutionPipeline
|
52
|
+
from .pipeline_if_superresolution import IFSuperResolutionPipeline
|
53
|
+
from .safety_checker import IFSafetyChecker
|
54
|
+
from .watermark import IFWatermarker
|