Spaces:
Running
on
Zero
Running
on
Zero
fix diffusers
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- diffusers/__init__.py +797 -0
- diffusers/commands/__init__.py +27 -0
- diffusers/experimental/__init__.py +1 -0
- diffusers/experimental/rl/__init__.py +1 -0
- diffusers/loaders/__init__.py +88 -0
- diffusers/models/__init__.py +105 -0
- diffusers/models/autoencoders/__init__.py +5 -0
- diffusers/models/transformers/__init__.py +9 -0
- diffusers/models/unets/__init__.py +18 -0
- diffusers/pipelines/__init__.py +591 -0
- diffusers/pipelines/amused/__init__.py +62 -0
- diffusers/pipelines/animatediff/__init__.py +49 -0
- diffusers/pipelines/audioldm/__init__.py +51 -0
- diffusers/pipelines/audioldm2/__init__.py +50 -0
- diffusers/pipelines/blip_diffusion/__init__.py +20 -0
- diffusers/pipelines/consistency_models/__init__.py +24 -0
- diffusers/pipelines/controlnet/__init__.py +80 -0
- diffusers/pipelines/controlnet_xs/__init__.py +68 -0
- diffusers/pipelines/dance_diffusion/__init__.py +18 -0
- diffusers/pipelines/ddim/__init__.py +18 -0
- diffusers/pipelines/ddpm/__init__.py +22 -0
- diffusers/pipelines/deepfloyd_if/__init__.py +85 -0
- diffusers/pipelines/deprecated/__init__.py +153 -0
- diffusers/pipelines/deprecated/alt_diffusion/__init__.py +53 -0
- diffusers/pipelines/deprecated/audio_diffusion/__init__.py +23 -0
- diffusers/pipelines/deprecated/latent_diffusion_uncond/__init__.py +18 -0
- diffusers/pipelines/deprecated/pndm/__init__.py +18 -0
- diffusers/pipelines/deprecated/repaint/__init__.py +19 -0
- diffusers/pipelines/deprecated/score_sde_ve/__init__.py +19 -0
- diffusers/pipelines/deprecated/spectrogram_diffusion/__init__.py +75 -0
- diffusers/pipelines/deprecated/stable_diffusion_variants/__init__.py +55 -0
- diffusers/pipelines/deprecated/stochastic_karras_ve/__init__.py +19 -0
- diffusers/pipelines/deprecated/versatile_diffusion/__init__.py +71 -0
- diffusers/pipelines/deprecated/vq_diffusion/__init__.py +57 -0
- diffusers/pipelines/dit/__init__.py +19 -0
- diffusers/pipelines/i2vgen_xl/__init__.py +46 -0
- diffusers/pipelines/kandinsky/__init__.py +66 -0
- diffusers/pipelines/kandinsky2_2/__init__.py +70 -0
- diffusers/pipelines/kandinsky3/__init__.py +49 -0
- diffusers/pipelines/latent_consistency_models/__init__.py +50 -0
- diffusers/pipelines/latent_diffusion/__init__.py +50 -0
- diffusers/pipelines/ledits_pp/__init__.py +55 -0
- diffusers/pipelines/musicldm/__init__.py +49 -0
- diffusers/pipelines/paint_by_example/__init__.py +55 -0
- diffusers/pipelines/pia/__init__.py +46 -0
- diffusers/pipelines/pixart_alpha/__init__.py +55 -0
- diffusers/pipelines/semantic_stable_diffusion/__init__.py +49 -0
- diffusers/pipelines/shap_e/__init__.py +71 -0
- diffusers/pipelines/stable_cascade/__init__.py +50 -0
- diffusers/pipelines/stable_diffusion/__init__.py +203 -0
diffusers/__init__.py
ADDED
@@ -0,0 +1,797 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
__version__ = "0.28.0.dev0"
|
2 |
+
|
3 |
+
from typing import TYPE_CHECKING
|
4 |
+
|
5 |
+
from .utils import (
|
6 |
+
DIFFUSERS_SLOW_IMPORT,
|
7 |
+
OptionalDependencyNotAvailable,
|
8 |
+
_LazyModule,
|
9 |
+
is_flax_available,
|
10 |
+
is_k_diffusion_available,
|
11 |
+
is_librosa_available,
|
12 |
+
is_note_seq_available,
|
13 |
+
is_onnx_available,
|
14 |
+
is_scipy_available,
|
15 |
+
is_torch_available,
|
16 |
+
is_torchsde_available,
|
17 |
+
is_transformers_available,
|
18 |
+
)
|
19 |
+
|
20 |
+
|
21 |
+
# Lazy Import based on
|
22 |
+
# https://github.com/huggingface/transformers/blob/main/src/transformers/__init__.py
|
23 |
+
|
24 |
+
# When adding a new object to this init, please add it to `_import_structure`. The `_import_structure` is a dictionary submodule to list of object names,
|
25 |
+
# and is used to defer the actual importing for when the objects are requested.
|
26 |
+
# This way `import diffusers` provides the names in the namespace without actually importing anything (and especially none of the backends).
|
27 |
+
|
28 |
+
_import_structure = {
|
29 |
+
"configuration_utils": ["ConfigMixin"],
|
30 |
+
"models": [],
|
31 |
+
"pipelines": [],
|
32 |
+
"schedulers": [],
|
33 |
+
"utils": [
|
34 |
+
"OptionalDependencyNotAvailable",
|
35 |
+
"is_flax_available",
|
36 |
+
"is_inflect_available",
|
37 |
+
"is_invisible_watermark_available",
|
38 |
+
"is_k_diffusion_available",
|
39 |
+
"is_k_diffusion_version",
|
40 |
+
"is_librosa_available",
|
41 |
+
"is_note_seq_available",
|
42 |
+
"is_onnx_available",
|
43 |
+
"is_scipy_available",
|
44 |
+
"is_torch_available",
|
45 |
+
"is_torchsde_available",
|
46 |
+
"is_transformers_available",
|
47 |
+
"is_transformers_version",
|
48 |
+
"is_unidecode_available",
|
49 |
+
"logging",
|
50 |
+
],
|
51 |
+
}
|
52 |
+
|
53 |
+
try:
|
54 |
+
if not is_onnx_available():
|
55 |
+
raise OptionalDependencyNotAvailable()
|
56 |
+
except OptionalDependencyNotAvailable:
|
57 |
+
from .utils import dummy_onnx_objects # noqa F403
|
58 |
+
|
59 |
+
_import_structure["utils.dummy_onnx_objects"] = [
|
60 |
+
name for name in dir(dummy_onnx_objects) if not name.startswith("_")
|
61 |
+
]
|
62 |
+
|
63 |
+
else:
|
64 |
+
_import_structure["pipelines"].extend(["OnnxRuntimeModel"])
|
65 |
+
|
66 |
+
try:
|
67 |
+
if not is_torch_available():
|
68 |
+
raise OptionalDependencyNotAvailable()
|
69 |
+
except OptionalDependencyNotAvailable:
|
70 |
+
from .utils import dummy_pt_objects # noqa F403
|
71 |
+
|
72 |
+
_import_structure["utils.dummy_pt_objects"] = [name for name in dir(dummy_pt_objects) if not name.startswith("_")]
|
73 |
+
|
74 |
+
else:
|
75 |
+
_import_structure["models"].extend(
|
76 |
+
[
|
77 |
+
"AsymmetricAutoencoderKL",
|
78 |
+
"AutoencoderKL",
|
79 |
+
"AutoencoderKLTemporalDecoder",
|
80 |
+
"AutoencoderTiny",
|
81 |
+
"ConsistencyDecoderVAE",
|
82 |
+
"ControlNetModel",
|
83 |
+
"ControlNetXSAdapter",
|
84 |
+
"I2VGenXLUNet",
|
85 |
+
"Kandinsky3UNet",
|
86 |
+
"ModelMixin",
|
87 |
+
"MotionAdapter",
|
88 |
+
"MultiAdapter",
|
89 |
+
"PriorTransformer",
|
90 |
+
"StableCascadeUNet",
|
91 |
+
"T2IAdapter",
|
92 |
+
"T5FilmDecoder",
|
93 |
+
"Transformer2DModel",
|
94 |
+
"UNet1DModel",
|
95 |
+
"UNet2DConditionModel",
|
96 |
+
"UNet2DModel",
|
97 |
+
"UNet3DConditionModel",
|
98 |
+
"UNetControlNetXSModel",
|
99 |
+
"UNetMotionModel",
|
100 |
+
"UNetSpatioTemporalConditionModel",
|
101 |
+
"UVit2DModel",
|
102 |
+
"VQModel",
|
103 |
+
]
|
104 |
+
)
|
105 |
+
|
106 |
+
_import_structure["optimization"] = [
|
107 |
+
"get_constant_schedule",
|
108 |
+
"get_constant_schedule_with_warmup",
|
109 |
+
"get_cosine_schedule_with_warmup",
|
110 |
+
"get_cosine_with_hard_restarts_schedule_with_warmup",
|
111 |
+
"get_linear_schedule_with_warmup",
|
112 |
+
"get_polynomial_decay_schedule_with_warmup",
|
113 |
+
"get_scheduler",
|
114 |
+
]
|
115 |
+
_import_structure["pipelines"].extend(
|
116 |
+
[
|
117 |
+
"AudioPipelineOutput",
|
118 |
+
"AutoPipelineForImage2Image",
|
119 |
+
"AutoPipelineForInpainting",
|
120 |
+
"AutoPipelineForText2Image",
|
121 |
+
"ConsistencyModelPipeline",
|
122 |
+
"DanceDiffusionPipeline",
|
123 |
+
"DDIMPipeline",
|
124 |
+
"DDPMPipeline",
|
125 |
+
"DiffusionPipeline",
|
126 |
+
"DiTPipeline",
|
127 |
+
"ImagePipelineOutput",
|
128 |
+
"KarrasVePipeline",
|
129 |
+
"LDMPipeline",
|
130 |
+
"LDMSuperResolutionPipeline",
|
131 |
+
"PNDMPipeline",
|
132 |
+
"RePaintPipeline",
|
133 |
+
"ScoreSdeVePipeline",
|
134 |
+
"StableDiffusionMixin",
|
135 |
+
]
|
136 |
+
)
|
137 |
+
_import_structure["schedulers"].extend(
|
138 |
+
[
|
139 |
+
"AmusedScheduler",
|
140 |
+
"CMStochasticIterativeScheduler",
|
141 |
+
"DDIMInverseScheduler",
|
142 |
+
"DDIMParallelScheduler",
|
143 |
+
"DDIMScheduler",
|
144 |
+
"DDPMParallelScheduler",
|
145 |
+
"DDPMScheduler",
|
146 |
+
"DDPMWuerstchenScheduler",
|
147 |
+
"DEISMultistepScheduler",
|
148 |
+
"DPMSolverMultistepInverseScheduler",
|
149 |
+
"DPMSolverMultistepScheduler",
|
150 |
+
"DPMSolverSinglestepScheduler",
|
151 |
+
"EDMDPMSolverMultistepScheduler",
|
152 |
+
"EDMEulerScheduler",
|
153 |
+
"EulerAncestralDiscreteScheduler",
|
154 |
+
"EulerDiscreteScheduler",
|
155 |
+
"HeunDiscreteScheduler",
|
156 |
+
"IPNDMScheduler",
|
157 |
+
"KarrasVeScheduler",
|
158 |
+
"KDPM2AncestralDiscreteScheduler",
|
159 |
+
"KDPM2DiscreteScheduler",
|
160 |
+
"LCMScheduler",
|
161 |
+
"PNDMScheduler",
|
162 |
+
"RePaintScheduler",
|
163 |
+
"SASolverScheduler",
|
164 |
+
"SchedulerMixin",
|
165 |
+
"ScoreSdeVeScheduler",
|
166 |
+
"TCDScheduler",
|
167 |
+
"UnCLIPScheduler",
|
168 |
+
"UniPCMultistepScheduler",
|
169 |
+
"VQDiffusionScheduler",
|
170 |
+
]
|
171 |
+
)
|
172 |
+
_import_structure["training_utils"] = ["EMAModel"]
|
173 |
+
|
174 |
+
try:
|
175 |
+
if not (is_torch_available() and is_scipy_available()):
|
176 |
+
raise OptionalDependencyNotAvailable()
|
177 |
+
except OptionalDependencyNotAvailable:
|
178 |
+
from .utils import dummy_torch_and_scipy_objects # noqa F403
|
179 |
+
|
180 |
+
_import_structure["utils.dummy_torch_and_scipy_objects"] = [
|
181 |
+
name for name in dir(dummy_torch_and_scipy_objects) if not name.startswith("_")
|
182 |
+
]
|
183 |
+
|
184 |
+
else:
|
185 |
+
_import_structure["schedulers"].extend(["LMSDiscreteScheduler"])
|
186 |
+
|
187 |
+
try:
|
188 |
+
if not (is_torch_available() and is_torchsde_available()):
|
189 |
+
raise OptionalDependencyNotAvailable()
|
190 |
+
except OptionalDependencyNotAvailable:
|
191 |
+
from .utils import dummy_torch_and_torchsde_objects # noqa F403
|
192 |
+
|
193 |
+
_import_structure["utils.dummy_torch_and_torchsde_objects"] = [
|
194 |
+
name for name in dir(dummy_torch_and_torchsde_objects) if not name.startswith("_")
|
195 |
+
]
|
196 |
+
|
197 |
+
else:
|
198 |
+
_import_structure["schedulers"].extend(["DPMSolverSDEScheduler"])
|
199 |
+
|
200 |
+
try:
|
201 |
+
if not (is_torch_available() and is_transformers_available()):
|
202 |
+
raise OptionalDependencyNotAvailable()
|
203 |
+
except OptionalDependencyNotAvailable:
|
204 |
+
from .utils import dummy_torch_and_transformers_objects # noqa F403
|
205 |
+
|
206 |
+
_import_structure["utils.dummy_torch_and_transformers_objects"] = [
|
207 |
+
name for name in dir(dummy_torch_and_transformers_objects) if not name.startswith("_")
|
208 |
+
]
|
209 |
+
|
210 |
+
else:
|
211 |
+
_import_structure["pipelines"].extend(
|
212 |
+
[
|
213 |
+
"AltDiffusionImg2ImgPipeline",
|
214 |
+
"AltDiffusionPipeline",
|
215 |
+
"AmusedImg2ImgPipeline",
|
216 |
+
"AmusedInpaintPipeline",
|
217 |
+
"AmusedPipeline",
|
218 |
+
"AnimateDiffPipeline",
|
219 |
+
"AnimateDiffVideoToVideoPipeline",
|
220 |
+
"AudioLDM2Pipeline",
|
221 |
+
"AudioLDM2ProjectionModel",
|
222 |
+
"AudioLDM2UNet2DConditionModel",
|
223 |
+
"AudioLDMPipeline",
|
224 |
+
"BlipDiffusionControlNetPipeline",
|
225 |
+
"BlipDiffusionPipeline",
|
226 |
+
"CLIPImageProjection",
|
227 |
+
"CycleDiffusionPipeline",
|
228 |
+
"I2VGenXLPipeline",
|
229 |
+
"IFImg2ImgPipeline",
|
230 |
+
"IFImg2ImgSuperResolutionPipeline",
|
231 |
+
"IFInpaintingPipeline",
|
232 |
+
"IFInpaintingSuperResolutionPipeline",
|
233 |
+
"IFPipeline",
|
234 |
+
"IFSuperResolutionPipeline",
|
235 |
+
"ImageTextPipelineOutput",
|
236 |
+
"Kandinsky3Img2ImgPipeline",
|
237 |
+
"Kandinsky3Pipeline",
|
238 |
+
"KandinskyCombinedPipeline",
|
239 |
+
"KandinskyImg2ImgCombinedPipeline",
|
240 |
+
"KandinskyImg2ImgPipeline",
|
241 |
+
"KandinskyInpaintCombinedPipeline",
|
242 |
+
"KandinskyInpaintPipeline",
|
243 |
+
"KandinskyPipeline",
|
244 |
+
"KandinskyPriorPipeline",
|
245 |
+
"KandinskyV22CombinedPipeline",
|
246 |
+
"KandinskyV22ControlnetImg2ImgPipeline",
|
247 |
+
"KandinskyV22ControlnetPipeline",
|
248 |
+
"KandinskyV22Img2ImgCombinedPipeline",
|
249 |
+
"KandinskyV22Img2ImgPipeline",
|
250 |
+
"KandinskyV22InpaintCombinedPipeline",
|
251 |
+
"KandinskyV22InpaintPipeline",
|
252 |
+
"KandinskyV22Pipeline",
|
253 |
+
"KandinskyV22PriorEmb2EmbPipeline",
|
254 |
+
"KandinskyV22PriorPipeline",
|
255 |
+
"LatentConsistencyModelImg2ImgPipeline",
|
256 |
+
"LatentConsistencyModelPipeline",
|
257 |
+
"LDMTextToImagePipeline",
|
258 |
+
"LEditsPPPipelineStableDiffusion",
|
259 |
+
"LEditsPPPipelineStableDiffusionXL",
|
260 |
+
"MusicLDMPipeline",
|
261 |
+
"PaintByExamplePipeline",
|
262 |
+
"PIAPipeline",
|
263 |
+
"PixArtAlphaPipeline",
|
264 |
+
"PixArtSigmaPipeline",
|
265 |
+
"SemanticStableDiffusionPipeline",
|
266 |
+
"ShapEImg2ImgPipeline",
|
267 |
+
"ShapEPipeline",
|
268 |
+
"StableCascadeCombinedPipeline",
|
269 |
+
"StableCascadeDecoderPipeline",
|
270 |
+
"StableCascadePriorPipeline",
|
271 |
+
"StableDiffusionAdapterPipeline",
|
272 |
+
"StableDiffusionAttendAndExcitePipeline",
|
273 |
+
"StableDiffusionControlNetImg2ImgPipeline",
|
274 |
+
"StableDiffusionControlNetInpaintPipeline",
|
275 |
+
"StableDiffusionControlNetPipeline",
|
276 |
+
"StableDiffusionControlNetXSPipeline",
|
277 |
+
"StableDiffusionDepth2ImgPipeline",
|
278 |
+
"StableDiffusionDiffEditPipeline",
|
279 |
+
"StableDiffusionGLIGENPipeline",
|
280 |
+
"StableDiffusionGLIGENTextImagePipeline",
|
281 |
+
"StableDiffusionImageVariationPipeline",
|
282 |
+
"StableDiffusionImg2ImgPipeline",
|
283 |
+
"StableDiffusionInpaintPipeline",
|
284 |
+
"StableDiffusionInpaintPipelineLegacy",
|
285 |
+
"StableDiffusionInstructPix2PixPipeline",
|
286 |
+
"StableDiffusionLatentUpscalePipeline",
|
287 |
+
"StableDiffusionLDM3DPipeline",
|
288 |
+
"StableDiffusionModelEditingPipeline",
|
289 |
+
"StableDiffusionPanoramaPipeline",
|
290 |
+
"StableDiffusionParadigmsPipeline",
|
291 |
+
"StableDiffusionPipeline",
|
292 |
+
"StableDiffusionPipelineSafe",
|
293 |
+
"StableDiffusionPix2PixZeroPipeline",
|
294 |
+
"StableDiffusionSAGPipeline",
|
295 |
+
"StableDiffusionUpscalePipeline",
|
296 |
+
"StableDiffusionXLAdapterPipeline",
|
297 |
+
"StableDiffusionXLControlNetImg2ImgPipeline",
|
298 |
+
"StableDiffusionXLControlNetInpaintPipeline",
|
299 |
+
"StableDiffusionXLControlNetPipeline",
|
300 |
+
"StableDiffusionXLControlNetXSPipeline",
|
301 |
+
"StableDiffusionXLImg2ImgPipeline",
|
302 |
+
"StableDiffusionXLInpaintPipeline",
|
303 |
+
"StableDiffusionXLInstructPix2PixPipeline",
|
304 |
+
"StableDiffusionXLPipeline",
|
305 |
+
"StableUnCLIPImg2ImgPipeline",
|
306 |
+
"StableUnCLIPPipeline",
|
307 |
+
"StableVideoDiffusionPipeline",
|
308 |
+
"TextToVideoSDPipeline",
|
309 |
+
"TextToVideoZeroPipeline",
|
310 |
+
"TextToVideoZeroSDXLPipeline",
|
311 |
+
"UnCLIPImageVariationPipeline",
|
312 |
+
"UnCLIPPipeline",
|
313 |
+
"UniDiffuserModel",
|
314 |
+
"UniDiffuserPipeline",
|
315 |
+
"UniDiffuserTextDecoder",
|
316 |
+
"VersatileDiffusionDualGuidedPipeline",
|
317 |
+
"VersatileDiffusionImageVariationPipeline",
|
318 |
+
"VersatileDiffusionPipeline",
|
319 |
+
"VersatileDiffusionTextToImagePipeline",
|
320 |
+
"VideoToVideoSDPipeline",
|
321 |
+
"VQDiffusionPipeline",
|
322 |
+
"WuerstchenCombinedPipeline",
|
323 |
+
"WuerstchenDecoderPipeline",
|
324 |
+
"WuerstchenPriorPipeline",
|
325 |
+
]
|
326 |
+
)
|
327 |
+
|
328 |
+
try:
|
329 |
+
if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
|
330 |
+
raise OptionalDependencyNotAvailable()
|
331 |
+
except OptionalDependencyNotAvailable:
|
332 |
+
from .utils import dummy_torch_and_transformers_and_k_diffusion_objects # noqa F403
|
333 |
+
|
334 |
+
_import_structure["utils.dummy_torch_and_transformers_and_k_diffusion_objects"] = [
|
335 |
+
name for name in dir(dummy_torch_and_transformers_and_k_diffusion_objects) if not name.startswith("_")
|
336 |
+
]
|
337 |
+
|
338 |
+
else:
|
339 |
+
_import_structure["pipelines"].extend(["StableDiffusionKDiffusionPipeline", "StableDiffusionXLKDiffusionPipeline"])
|
340 |
+
|
341 |
+
try:
|
342 |
+
if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
|
343 |
+
raise OptionalDependencyNotAvailable()
|
344 |
+
except OptionalDependencyNotAvailable:
|
345 |
+
from .utils import dummy_torch_and_transformers_and_onnx_objects # noqa F403
|
346 |
+
|
347 |
+
_import_structure["utils.dummy_torch_and_transformers_and_onnx_objects"] = [
|
348 |
+
name for name in dir(dummy_torch_and_transformers_and_onnx_objects) if not name.startswith("_")
|
349 |
+
]
|
350 |
+
|
351 |
+
else:
|
352 |
+
_import_structure["pipelines"].extend(
|
353 |
+
[
|
354 |
+
"OnnxStableDiffusionImg2ImgPipeline",
|
355 |
+
"OnnxStableDiffusionInpaintPipeline",
|
356 |
+
"OnnxStableDiffusionInpaintPipelineLegacy",
|
357 |
+
"OnnxStableDiffusionPipeline",
|
358 |
+
"OnnxStableDiffusionUpscalePipeline",
|
359 |
+
"StableDiffusionOnnxPipeline",
|
360 |
+
]
|
361 |
+
)
|
362 |
+
|
363 |
+
try:
|
364 |
+
if not (is_torch_available() and is_librosa_available()):
|
365 |
+
raise OptionalDependencyNotAvailable()
|
366 |
+
except OptionalDependencyNotAvailable:
|
367 |
+
from .utils import dummy_torch_and_librosa_objects # noqa F403
|
368 |
+
|
369 |
+
_import_structure["utils.dummy_torch_and_librosa_objects"] = [
|
370 |
+
name for name in dir(dummy_torch_and_librosa_objects) if not name.startswith("_")
|
371 |
+
]
|
372 |
+
|
373 |
+
else:
|
374 |
+
_import_structure["pipelines"].extend(["AudioDiffusionPipeline", "Mel"])
|
375 |
+
|
376 |
+
try:
|
377 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
378 |
+
raise OptionalDependencyNotAvailable()
|
379 |
+
except OptionalDependencyNotAvailable:
|
380 |
+
from .utils import dummy_transformers_and_torch_and_note_seq_objects # noqa F403
|
381 |
+
|
382 |
+
_import_structure["utils.dummy_transformers_and_torch_and_note_seq_objects"] = [
|
383 |
+
name for name in dir(dummy_transformers_and_torch_and_note_seq_objects) if not name.startswith("_")
|
384 |
+
]
|
385 |
+
|
386 |
+
|
387 |
+
else:
|
388 |
+
_import_structure["pipelines"].extend(["SpectrogramDiffusionPipeline"])
|
389 |
+
|
390 |
+
try:
|
391 |
+
if not is_flax_available():
|
392 |
+
raise OptionalDependencyNotAvailable()
|
393 |
+
except OptionalDependencyNotAvailable:
|
394 |
+
from .utils import dummy_flax_objects # noqa F403
|
395 |
+
|
396 |
+
_import_structure["utils.dummy_flax_objects"] = [
|
397 |
+
name for name in dir(dummy_flax_objects) if not name.startswith("_")
|
398 |
+
]
|
399 |
+
|
400 |
+
|
401 |
+
else:
|
402 |
+
_import_structure["models.controlnet_flax"] = ["FlaxControlNetModel"]
|
403 |
+
_import_structure["models.modeling_flax_utils"] = ["FlaxModelMixin"]
|
404 |
+
_import_structure["models.unets.unet_2d_condition_flax"] = ["FlaxUNet2DConditionModel"]
|
405 |
+
_import_structure["models.vae_flax"] = ["FlaxAutoencoderKL"]
|
406 |
+
_import_structure["pipelines"].extend(["FlaxDiffusionPipeline"])
|
407 |
+
_import_structure["schedulers"].extend(
|
408 |
+
[
|
409 |
+
"FlaxDDIMScheduler",
|
410 |
+
"FlaxDDPMScheduler",
|
411 |
+
"FlaxDPMSolverMultistepScheduler",
|
412 |
+
"FlaxEulerDiscreteScheduler",
|
413 |
+
"FlaxKarrasVeScheduler",
|
414 |
+
"FlaxLMSDiscreteScheduler",
|
415 |
+
"FlaxPNDMScheduler",
|
416 |
+
"FlaxSchedulerMixin",
|
417 |
+
"FlaxScoreSdeVeScheduler",
|
418 |
+
]
|
419 |
+
)
|
420 |
+
|
421 |
+
|
422 |
+
try:
|
423 |
+
if not (is_flax_available() and is_transformers_available()):
|
424 |
+
raise OptionalDependencyNotAvailable()
|
425 |
+
except OptionalDependencyNotAvailable:
|
426 |
+
from .utils import dummy_flax_and_transformers_objects # noqa F403
|
427 |
+
|
428 |
+
_import_structure["utils.dummy_flax_and_transformers_objects"] = [
|
429 |
+
name for name in dir(dummy_flax_and_transformers_objects) if not name.startswith("_")
|
430 |
+
]
|
431 |
+
|
432 |
+
|
433 |
+
else:
|
434 |
+
_import_structure["pipelines"].extend(
|
435 |
+
[
|
436 |
+
"FlaxStableDiffusionControlNetPipeline",
|
437 |
+
"FlaxStableDiffusionImg2ImgPipeline",
|
438 |
+
"FlaxStableDiffusionInpaintPipeline",
|
439 |
+
"FlaxStableDiffusionPipeline",
|
440 |
+
"FlaxStableDiffusionXLPipeline",
|
441 |
+
]
|
442 |
+
)
|
443 |
+
|
444 |
+
try:
|
445 |
+
if not (is_note_seq_available()):
|
446 |
+
raise OptionalDependencyNotAvailable()
|
447 |
+
except OptionalDependencyNotAvailable:
|
448 |
+
from .utils import dummy_note_seq_objects # noqa F403
|
449 |
+
|
450 |
+
_import_structure["utils.dummy_note_seq_objects"] = [
|
451 |
+
name for name in dir(dummy_note_seq_objects) if not name.startswith("_")
|
452 |
+
]
|
453 |
+
|
454 |
+
|
455 |
+
else:
|
456 |
+
_import_structure["pipelines"].extend(["MidiProcessor"])
|
457 |
+
|
458 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
459 |
+
from .configuration_utils import ConfigMixin
|
460 |
+
|
461 |
+
try:
|
462 |
+
if not is_onnx_available():
|
463 |
+
raise OptionalDependencyNotAvailable()
|
464 |
+
except OptionalDependencyNotAvailable:
|
465 |
+
from .utils.dummy_onnx_objects import * # noqa F403
|
466 |
+
else:
|
467 |
+
from .pipelines import OnnxRuntimeModel
|
468 |
+
|
469 |
+
try:
|
470 |
+
if not is_torch_available():
|
471 |
+
raise OptionalDependencyNotAvailable()
|
472 |
+
except OptionalDependencyNotAvailable:
|
473 |
+
from .utils.dummy_pt_objects import * # noqa F403
|
474 |
+
else:
|
475 |
+
from .models import (
|
476 |
+
AsymmetricAutoencoderKL,
|
477 |
+
AutoencoderKL,
|
478 |
+
AutoencoderKLTemporalDecoder,
|
479 |
+
AutoencoderTiny,
|
480 |
+
ConsistencyDecoderVAE,
|
481 |
+
ControlNetModel,
|
482 |
+
ControlNetXSAdapter,
|
483 |
+
I2VGenXLUNet,
|
484 |
+
Kandinsky3UNet,
|
485 |
+
ModelMixin,
|
486 |
+
MotionAdapter,
|
487 |
+
MultiAdapter,
|
488 |
+
PriorTransformer,
|
489 |
+
T2IAdapter,
|
490 |
+
T5FilmDecoder,
|
491 |
+
Transformer2DModel,
|
492 |
+
UNet1DModel,
|
493 |
+
UNet2DConditionModel,
|
494 |
+
UNet2DModel,
|
495 |
+
UNet3DConditionModel,
|
496 |
+
UNetControlNetXSModel,
|
497 |
+
UNetMotionModel,
|
498 |
+
UNetSpatioTemporalConditionModel,
|
499 |
+
UVit2DModel,
|
500 |
+
VQModel,
|
501 |
+
)
|
502 |
+
from .optimization import (
|
503 |
+
get_constant_schedule,
|
504 |
+
get_constant_schedule_with_warmup,
|
505 |
+
get_cosine_schedule_with_warmup,
|
506 |
+
get_cosine_with_hard_restarts_schedule_with_warmup,
|
507 |
+
get_linear_schedule_with_warmup,
|
508 |
+
get_polynomial_decay_schedule_with_warmup,
|
509 |
+
get_scheduler,
|
510 |
+
)
|
511 |
+
from .pipelines import (
|
512 |
+
AudioPipelineOutput,
|
513 |
+
AutoPipelineForImage2Image,
|
514 |
+
AutoPipelineForInpainting,
|
515 |
+
AutoPipelineForText2Image,
|
516 |
+
BlipDiffusionControlNetPipeline,
|
517 |
+
BlipDiffusionPipeline,
|
518 |
+
CLIPImageProjection,
|
519 |
+
ConsistencyModelPipeline,
|
520 |
+
DanceDiffusionPipeline,
|
521 |
+
DDIMPipeline,
|
522 |
+
DDPMPipeline,
|
523 |
+
DiffusionPipeline,
|
524 |
+
DiTPipeline,
|
525 |
+
ImagePipelineOutput,
|
526 |
+
KarrasVePipeline,
|
527 |
+
LDMPipeline,
|
528 |
+
LDMSuperResolutionPipeline,
|
529 |
+
PNDMPipeline,
|
530 |
+
RePaintPipeline,
|
531 |
+
ScoreSdeVePipeline,
|
532 |
+
StableDiffusionMixin,
|
533 |
+
)
|
534 |
+
from .schedulers import (
|
535 |
+
AmusedScheduler,
|
536 |
+
CMStochasticIterativeScheduler,
|
537 |
+
DDIMInverseScheduler,
|
538 |
+
DDIMParallelScheduler,
|
539 |
+
DDIMScheduler,
|
540 |
+
DDPMParallelScheduler,
|
541 |
+
DDPMScheduler,
|
542 |
+
DDPMWuerstchenScheduler,
|
543 |
+
DEISMultistepScheduler,
|
544 |
+
DPMSolverMultistepInverseScheduler,
|
545 |
+
DPMSolverMultistepScheduler,
|
546 |
+
DPMSolverSinglestepScheduler,
|
547 |
+
EDMDPMSolverMultistepScheduler,
|
548 |
+
EDMEulerScheduler,
|
549 |
+
EulerAncestralDiscreteScheduler,
|
550 |
+
EulerDiscreteScheduler,
|
551 |
+
HeunDiscreteScheduler,
|
552 |
+
IPNDMScheduler,
|
553 |
+
KarrasVeScheduler,
|
554 |
+
KDPM2AncestralDiscreteScheduler,
|
555 |
+
KDPM2DiscreteScheduler,
|
556 |
+
LCMScheduler,
|
557 |
+
PNDMScheduler,
|
558 |
+
RePaintScheduler,
|
559 |
+
SASolverScheduler,
|
560 |
+
SchedulerMixin,
|
561 |
+
ScoreSdeVeScheduler,
|
562 |
+
TCDScheduler,
|
563 |
+
UnCLIPScheduler,
|
564 |
+
UniPCMultistepScheduler,
|
565 |
+
VQDiffusionScheduler,
|
566 |
+
)
|
567 |
+
from .training_utils import EMAModel
|
568 |
+
|
569 |
+
try:
|
570 |
+
if not (is_torch_available() and is_scipy_available()):
|
571 |
+
raise OptionalDependencyNotAvailable()
|
572 |
+
except OptionalDependencyNotAvailable:
|
573 |
+
from .utils.dummy_torch_and_scipy_objects import * # noqa F403
|
574 |
+
else:
|
575 |
+
from .schedulers import LMSDiscreteScheduler
|
576 |
+
|
577 |
+
try:
|
578 |
+
if not (is_torch_available() and is_torchsde_available()):
|
579 |
+
raise OptionalDependencyNotAvailable()
|
580 |
+
except OptionalDependencyNotAvailable:
|
581 |
+
from .utils.dummy_torch_and_torchsde_objects import * # noqa F403
|
582 |
+
else:
|
583 |
+
from .schedulers import DPMSolverSDEScheduler
|
584 |
+
|
585 |
+
try:
|
586 |
+
if not (is_torch_available() and is_transformers_available()):
|
587 |
+
raise OptionalDependencyNotAvailable()
|
588 |
+
except OptionalDependencyNotAvailable:
|
589 |
+
from .utils.dummy_torch_and_transformers_objects import * # noqa F403
|
590 |
+
else:
|
591 |
+
from .pipelines import (
|
592 |
+
AltDiffusionImg2ImgPipeline,
|
593 |
+
AltDiffusionPipeline,
|
594 |
+
AmusedImg2ImgPipeline,
|
595 |
+
AmusedInpaintPipeline,
|
596 |
+
AmusedPipeline,
|
597 |
+
AnimateDiffPipeline,
|
598 |
+
AnimateDiffVideoToVideoPipeline,
|
599 |
+
AudioLDM2Pipeline,
|
600 |
+
AudioLDM2ProjectionModel,
|
601 |
+
AudioLDM2UNet2DConditionModel,
|
602 |
+
AudioLDMPipeline,
|
603 |
+
CLIPImageProjection,
|
604 |
+
CycleDiffusionPipeline,
|
605 |
+
I2VGenXLPipeline,
|
606 |
+
IFImg2ImgPipeline,
|
607 |
+
IFImg2ImgSuperResolutionPipeline,
|
608 |
+
IFInpaintingPipeline,
|
609 |
+
IFInpaintingSuperResolutionPipeline,
|
610 |
+
IFPipeline,
|
611 |
+
IFSuperResolutionPipeline,
|
612 |
+
ImageTextPipelineOutput,
|
613 |
+
Kandinsky3Img2ImgPipeline,
|
614 |
+
Kandinsky3Pipeline,
|
615 |
+
KandinskyCombinedPipeline,
|
616 |
+
KandinskyImg2ImgCombinedPipeline,
|
617 |
+
KandinskyImg2ImgPipeline,
|
618 |
+
KandinskyInpaintCombinedPipeline,
|
619 |
+
KandinskyInpaintPipeline,
|
620 |
+
KandinskyPipeline,
|
621 |
+
KandinskyPriorPipeline,
|
622 |
+
KandinskyV22CombinedPipeline,
|
623 |
+
KandinskyV22ControlnetImg2ImgPipeline,
|
624 |
+
KandinskyV22ControlnetPipeline,
|
625 |
+
KandinskyV22Img2ImgCombinedPipeline,
|
626 |
+
KandinskyV22Img2ImgPipeline,
|
627 |
+
KandinskyV22InpaintCombinedPipeline,
|
628 |
+
KandinskyV22InpaintPipeline,
|
629 |
+
KandinskyV22Pipeline,
|
630 |
+
KandinskyV22PriorEmb2EmbPipeline,
|
631 |
+
KandinskyV22PriorPipeline,
|
632 |
+
LatentConsistencyModelImg2ImgPipeline,
|
633 |
+
LatentConsistencyModelPipeline,
|
634 |
+
LDMTextToImagePipeline,
|
635 |
+
LEditsPPPipelineStableDiffusion,
|
636 |
+
LEditsPPPipelineStableDiffusionXL,
|
637 |
+
MusicLDMPipeline,
|
638 |
+
PaintByExamplePipeline,
|
639 |
+
PIAPipeline,
|
640 |
+
PixArtAlphaPipeline,
|
641 |
+
PixArtSigmaPipeline,
|
642 |
+
SemanticStableDiffusionPipeline,
|
643 |
+
ShapEImg2ImgPipeline,
|
644 |
+
ShapEPipeline,
|
645 |
+
StableCascadeCombinedPipeline,
|
646 |
+
StableCascadeDecoderPipeline,
|
647 |
+
StableCascadePriorPipeline,
|
648 |
+
StableDiffusionAdapterPipeline,
|
649 |
+
StableDiffusionAttendAndExcitePipeline,
|
650 |
+
StableDiffusionControlNetImg2ImgPipeline,
|
651 |
+
StableDiffusionControlNetInpaintPipeline,
|
652 |
+
StableDiffusionControlNetPipeline,
|
653 |
+
StableDiffusionControlNetXSPipeline,
|
654 |
+
StableDiffusionDepth2ImgPipeline,
|
655 |
+
StableDiffusionDiffEditPipeline,
|
656 |
+
StableDiffusionGLIGENPipeline,
|
657 |
+
StableDiffusionGLIGENTextImagePipeline,
|
658 |
+
StableDiffusionImageVariationPipeline,
|
659 |
+
StableDiffusionImg2ImgPipeline,
|
660 |
+
StableDiffusionInpaintPipeline,
|
661 |
+
StableDiffusionInpaintPipelineLegacy,
|
662 |
+
StableDiffusionInstructPix2PixPipeline,
|
663 |
+
StableDiffusionLatentUpscalePipeline,
|
664 |
+
StableDiffusionLDM3DPipeline,
|
665 |
+
StableDiffusionModelEditingPipeline,
|
666 |
+
StableDiffusionPanoramaPipeline,
|
667 |
+
StableDiffusionParadigmsPipeline,
|
668 |
+
StableDiffusionPipeline,
|
669 |
+
StableDiffusionPipelineSafe,
|
670 |
+
StableDiffusionPix2PixZeroPipeline,
|
671 |
+
StableDiffusionSAGPipeline,
|
672 |
+
StableDiffusionUpscalePipeline,
|
673 |
+
StableDiffusionXLAdapterPipeline,
|
674 |
+
StableDiffusionXLControlNetImg2ImgPipeline,
|
675 |
+
StableDiffusionXLControlNetInpaintPipeline,
|
676 |
+
StableDiffusionXLControlNetPipeline,
|
677 |
+
StableDiffusionXLControlNetXSPipeline,
|
678 |
+
StableDiffusionXLImg2ImgPipeline,
|
679 |
+
StableDiffusionXLInpaintPipeline,
|
680 |
+
StableDiffusionXLInstructPix2PixPipeline,
|
681 |
+
StableDiffusionXLPipeline,
|
682 |
+
StableUnCLIPImg2ImgPipeline,
|
683 |
+
StableUnCLIPPipeline,
|
684 |
+
StableVideoDiffusionPipeline,
|
685 |
+
TextToVideoSDPipeline,
|
686 |
+
TextToVideoZeroPipeline,
|
687 |
+
TextToVideoZeroSDXLPipeline,
|
688 |
+
UnCLIPImageVariationPipeline,
|
689 |
+
UnCLIPPipeline,
|
690 |
+
UniDiffuserModel,
|
691 |
+
UniDiffuserPipeline,
|
692 |
+
UniDiffuserTextDecoder,
|
693 |
+
VersatileDiffusionDualGuidedPipeline,
|
694 |
+
VersatileDiffusionImageVariationPipeline,
|
695 |
+
VersatileDiffusionPipeline,
|
696 |
+
VersatileDiffusionTextToImagePipeline,
|
697 |
+
VideoToVideoSDPipeline,
|
698 |
+
VQDiffusionPipeline,
|
699 |
+
WuerstchenCombinedPipeline,
|
700 |
+
WuerstchenDecoderPipeline,
|
701 |
+
WuerstchenPriorPipeline,
|
702 |
+
)
|
703 |
+
|
704 |
+
try:
|
705 |
+
if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
|
706 |
+
raise OptionalDependencyNotAvailable()
|
707 |
+
except OptionalDependencyNotAvailable:
|
708 |
+
from .utils.dummy_torch_and_transformers_and_k_diffusion_objects import * # noqa F403
|
709 |
+
else:
|
710 |
+
from .pipelines import StableDiffusionKDiffusionPipeline, StableDiffusionXLKDiffusionPipeline
|
711 |
+
|
712 |
+
try:
|
713 |
+
if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
|
714 |
+
raise OptionalDependencyNotAvailable()
|
715 |
+
except OptionalDependencyNotAvailable:
|
716 |
+
from .utils.dummy_torch_and_transformers_and_onnx_objects import * # noqa F403
|
717 |
+
else:
|
718 |
+
from .pipelines import (
|
719 |
+
OnnxStableDiffusionImg2ImgPipeline,
|
720 |
+
OnnxStableDiffusionInpaintPipeline,
|
721 |
+
OnnxStableDiffusionInpaintPipelineLegacy,
|
722 |
+
OnnxStableDiffusionPipeline,
|
723 |
+
OnnxStableDiffusionUpscalePipeline,
|
724 |
+
StableDiffusionOnnxPipeline,
|
725 |
+
)
|
726 |
+
|
727 |
+
try:
|
728 |
+
if not (is_torch_available() and is_librosa_available()):
|
729 |
+
raise OptionalDependencyNotAvailable()
|
730 |
+
except OptionalDependencyNotAvailable:
|
731 |
+
from .utils.dummy_torch_and_librosa_objects import * # noqa F403
|
732 |
+
else:
|
733 |
+
from .pipelines import AudioDiffusionPipeline, Mel
|
734 |
+
|
735 |
+
try:
|
736 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
737 |
+
raise OptionalDependencyNotAvailable()
|
738 |
+
except OptionalDependencyNotAvailable:
|
739 |
+
from .utils.dummy_transformers_and_torch_and_note_seq_objects import * # noqa F403
|
740 |
+
else:
|
741 |
+
from .pipelines import SpectrogramDiffusionPipeline
|
742 |
+
|
743 |
+
try:
|
744 |
+
if not is_flax_available():
|
745 |
+
raise OptionalDependencyNotAvailable()
|
746 |
+
except OptionalDependencyNotAvailable:
|
747 |
+
from .utils.dummy_flax_objects import * # noqa F403
|
748 |
+
else:
|
749 |
+
from .models.controlnet_flax import FlaxControlNetModel
|
750 |
+
from .models.modeling_flax_utils import FlaxModelMixin
|
751 |
+
from .models.unets.unet_2d_condition_flax import FlaxUNet2DConditionModel
|
752 |
+
from .models.vae_flax import FlaxAutoencoderKL
|
753 |
+
from .pipelines import FlaxDiffusionPipeline
|
754 |
+
from .schedulers import (
|
755 |
+
FlaxDDIMScheduler,
|
756 |
+
FlaxDDPMScheduler,
|
757 |
+
FlaxDPMSolverMultistepScheduler,
|
758 |
+
FlaxEulerDiscreteScheduler,
|
759 |
+
FlaxKarrasVeScheduler,
|
760 |
+
FlaxLMSDiscreteScheduler,
|
761 |
+
FlaxPNDMScheduler,
|
762 |
+
FlaxSchedulerMixin,
|
763 |
+
FlaxScoreSdeVeScheduler,
|
764 |
+
)
|
765 |
+
|
766 |
+
try:
|
767 |
+
if not (is_flax_available() and is_transformers_available()):
|
768 |
+
raise OptionalDependencyNotAvailable()
|
769 |
+
except OptionalDependencyNotAvailable:
|
770 |
+
from .utils.dummy_flax_and_transformers_objects import * # noqa F403
|
771 |
+
else:
|
772 |
+
from .pipelines import (
|
773 |
+
FlaxStableDiffusionControlNetPipeline,
|
774 |
+
FlaxStableDiffusionImg2ImgPipeline,
|
775 |
+
FlaxStableDiffusionInpaintPipeline,
|
776 |
+
FlaxStableDiffusionPipeline,
|
777 |
+
FlaxStableDiffusionXLPipeline,
|
778 |
+
)
|
779 |
+
|
780 |
+
try:
|
781 |
+
if not (is_note_seq_available()):
|
782 |
+
raise OptionalDependencyNotAvailable()
|
783 |
+
except OptionalDependencyNotAvailable:
|
784 |
+
from .utils.dummy_note_seq_objects import * # noqa F403
|
785 |
+
else:
|
786 |
+
from .pipelines import MidiProcessor
|
787 |
+
|
788 |
+
else:
|
789 |
+
import sys
|
790 |
+
|
791 |
+
sys.modules[__name__] = _LazyModule(
|
792 |
+
__name__,
|
793 |
+
globals()["__file__"],
|
794 |
+
_import_structure,
|
795 |
+
module_spec=__spec__,
|
796 |
+
extra_objects={"__version__": __version__},
|
797 |
+
)
|
diffusers/commands/__init__.py
ADDED
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Copyright 2024 The HuggingFace Team. All rights reserved.
|
2 |
+
#
|
3 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4 |
+
# you may not use this file except in compliance with the License.
|
5 |
+
# You may obtain a copy of the License at
|
6 |
+
#
|
7 |
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8 |
+
#
|
9 |
+
# Unless required by applicable law or agreed to in writing, software
|
10 |
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11 |
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 |
+
# See the License for the specific language governing permissions and
|
13 |
+
# limitations under the License.
|
14 |
+
|
15 |
+
from abc import ABC, abstractmethod
|
16 |
+
from argparse import ArgumentParser
|
17 |
+
|
18 |
+
|
19 |
+
class BaseDiffusersCLICommand(ABC):
|
20 |
+
@staticmethod
|
21 |
+
@abstractmethod
|
22 |
+
def register_subcommand(parser: ArgumentParser):
|
23 |
+
raise NotImplementedError()
|
24 |
+
|
25 |
+
@abstractmethod
|
26 |
+
def run(self):
|
27 |
+
raise NotImplementedError()
|
diffusers/experimental/__init__.py
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
from .rl import ValueGuidedRLPipeline
|
diffusers/experimental/rl/__init__.py
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
from .value_guided_sampling import ValueGuidedRLPipeline
|
diffusers/loaders/__init__.py
ADDED
@@ -0,0 +1,88 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ..utils import DIFFUSERS_SLOW_IMPORT, _LazyModule, deprecate
|
4 |
+
from ..utils.import_utils import is_peft_available, is_torch_available, is_transformers_available
|
5 |
+
|
6 |
+
|
7 |
+
def text_encoder_lora_state_dict(text_encoder):
|
8 |
+
deprecate(
|
9 |
+
"text_encoder_load_state_dict in `models`",
|
10 |
+
"0.27.0",
|
11 |
+
"`text_encoder_lora_state_dict` is deprecated and will be removed in 0.27.0. Make sure to retrieve the weights using `get_peft_model`. See https://huggingface.co/docs/peft/v0.6.2/en/quicktour#peftmodel for more information.",
|
12 |
+
)
|
13 |
+
state_dict = {}
|
14 |
+
|
15 |
+
for name, module in text_encoder_attn_modules(text_encoder):
|
16 |
+
for k, v in module.q_proj.lora_linear_layer.state_dict().items():
|
17 |
+
state_dict[f"{name}.q_proj.lora_linear_layer.{k}"] = v
|
18 |
+
|
19 |
+
for k, v in module.k_proj.lora_linear_layer.state_dict().items():
|
20 |
+
state_dict[f"{name}.k_proj.lora_linear_layer.{k}"] = v
|
21 |
+
|
22 |
+
for k, v in module.v_proj.lora_linear_layer.state_dict().items():
|
23 |
+
state_dict[f"{name}.v_proj.lora_linear_layer.{k}"] = v
|
24 |
+
|
25 |
+
for k, v in module.out_proj.lora_linear_layer.state_dict().items():
|
26 |
+
state_dict[f"{name}.out_proj.lora_linear_layer.{k}"] = v
|
27 |
+
|
28 |
+
return state_dict
|
29 |
+
|
30 |
+
|
31 |
+
if is_transformers_available():
|
32 |
+
|
33 |
+
def text_encoder_attn_modules(text_encoder):
|
34 |
+
deprecate(
|
35 |
+
"text_encoder_attn_modules in `models`",
|
36 |
+
"0.27.0",
|
37 |
+
"`text_encoder_lora_state_dict` is deprecated and will be removed in 0.27.0. Make sure to retrieve the weights using `get_peft_model`. See https://huggingface.co/docs/peft/v0.6.2/en/quicktour#peftmodel for more information.",
|
38 |
+
)
|
39 |
+
from transformers import CLIPTextModel, CLIPTextModelWithProjection
|
40 |
+
|
41 |
+
attn_modules = []
|
42 |
+
|
43 |
+
if isinstance(text_encoder, (CLIPTextModel, CLIPTextModelWithProjection)):
|
44 |
+
for i, layer in enumerate(text_encoder.text_model.encoder.layers):
|
45 |
+
name = f"text_model.encoder.layers.{i}.self_attn"
|
46 |
+
mod = layer.self_attn
|
47 |
+
attn_modules.append((name, mod))
|
48 |
+
else:
|
49 |
+
raise ValueError(f"do not know how to get attention modules for: {text_encoder.__class__.__name__}")
|
50 |
+
|
51 |
+
return attn_modules
|
52 |
+
|
53 |
+
|
54 |
+
_import_structure = {}
|
55 |
+
|
56 |
+
if is_torch_available():
|
57 |
+
_import_structure["autoencoder"] = ["FromOriginalVAEMixin"]
|
58 |
+
|
59 |
+
_import_structure["controlnet"] = ["FromOriginalControlNetMixin"]
|
60 |
+
_import_structure["unet"] = ["UNet2DConditionLoadersMixin"]
|
61 |
+
_import_structure["utils"] = ["AttnProcsLayers"]
|
62 |
+
if is_transformers_available():
|
63 |
+
_import_structure["single_file"] = ["FromSingleFileMixin"]
|
64 |
+
_import_structure["lora"] = ["LoraLoaderMixin", "StableDiffusionXLLoraLoaderMixin"]
|
65 |
+
_import_structure["textual_inversion"] = ["TextualInversionLoaderMixin"]
|
66 |
+
_import_structure["ip_adapter"] = ["IPAdapterMixin"]
|
67 |
+
|
68 |
+
_import_structure["peft"] = ["PeftAdapterMixin"]
|
69 |
+
|
70 |
+
|
71 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
72 |
+
if is_torch_available():
|
73 |
+
from .autoencoder import FromOriginalVAEMixin
|
74 |
+
from .controlnet import FromOriginalControlNetMixin
|
75 |
+
from .unet import UNet2DConditionLoadersMixin
|
76 |
+
from .utils import AttnProcsLayers
|
77 |
+
|
78 |
+
if is_transformers_available():
|
79 |
+
from .ip_adapter import IPAdapterMixin
|
80 |
+
from .lora import LoraLoaderMixin, StableDiffusionXLLoraLoaderMixin
|
81 |
+
from .single_file import FromSingleFileMixin
|
82 |
+
from .textual_inversion import TextualInversionLoaderMixin
|
83 |
+
|
84 |
+
from .peft import PeftAdapterMixin
|
85 |
+
else:
|
86 |
+
import sys
|
87 |
+
|
88 |
+
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)
|
diffusers/models/__init__.py
ADDED
@@ -0,0 +1,105 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# Copyright 2024 The HuggingFace Team. All rights reserved.
|
2 |
+
#
|
3 |
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
4 |
+
# you may not use this file except in compliance with the License.
|
5 |
+
# You may obtain a copy of the License at
|
6 |
+
#
|
7 |
+
# http://www.apache.org/licenses/LICENSE-2.0
|
8 |
+
#
|
9 |
+
# Unless required by applicable law or agreed to in writing, software
|
10 |
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
11 |
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
12 |
+
# See the License for the specific language governing permissions and
|
13 |
+
# limitations under the License.
|
14 |
+
|
15 |
+
from typing import TYPE_CHECKING
|
16 |
+
|
17 |
+
from ..utils import (
|
18 |
+
DIFFUSERS_SLOW_IMPORT,
|
19 |
+
_LazyModule,
|
20 |
+
is_flax_available,
|
21 |
+
is_torch_available,
|
22 |
+
)
|
23 |
+
|
24 |
+
|
25 |
+
_import_structure = {}
|
26 |
+
|
27 |
+
if is_torch_available():
|
28 |
+
_import_structure["adapter"] = ["MultiAdapter", "T2IAdapter"]
|
29 |
+
_import_structure["autoencoders.autoencoder_asym_kl"] = ["AsymmetricAutoencoderKL"]
|
30 |
+
_import_structure["autoencoders.autoencoder_kl"] = ["AutoencoderKL"]
|
31 |
+
_import_structure["autoencoders.autoencoder_kl_temporal_decoder"] = ["AutoencoderKLTemporalDecoder"]
|
32 |
+
_import_structure["autoencoders.autoencoder_tiny"] = ["AutoencoderTiny"]
|
33 |
+
_import_structure["autoencoders.consistency_decoder_vae"] = ["ConsistencyDecoderVAE"]
|
34 |
+
_import_structure["controlnet"] = ["ControlNetModel"]
|
35 |
+
_import_structure["controlnet_xs"] = ["ControlNetXSAdapter", "UNetControlNetXSModel"]
|
36 |
+
_import_structure["dual_transformer_2d"] = ["DualTransformer2DModel"]
|
37 |
+
_import_structure["embeddings"] = ["ImageProjection"]
|
38 |
+
_import_structure["modeling_utils"] = ["ModelMixin"]
|
39 |
+
_import_structure["transformers.prior_transformer"] = ["PriorTransformer"]
|
40 |
+
_import_structure["transformers.t5_film_transformer"] = ["T5FilmDecoder"]
|
41 |
+
_import_structure["transformers.transformer_2d"] = ["Transformer2DModel"]
|
42 |
+
_import_structure["transformers.transformer_temporal"] = ["TransformerTemporalModel"]
|
43 |
+
_import_structure["unets.unet_1d"] = ["UNet1DModel"]
|
44 |
+
_import_structure["unets.unet_2d"] = ["UNet2DModel"]
|
45 |
+
_import_structure["unets.unet_2d_condition"] = ["UNet2DConditionModel"]
|
46 |
+
_import_structure["unets.unet_3d_condition"] = ["UNet3DConditionModel"]
|
47 |
+
_import_structure["unets.unet_i2vgen_xl"] = ["I2VGenXLUNet"]
|
48 |
+
_import_structure["unets.unet_kandinsky3"] = ["Kandinsky3UNet"]
|
49 |
+
_import_structure["unets.unet_motion_model"] = ["MotionAdapter", "UNetMotionModel"]
|
50 |
+
_import_structure["unets.unet_spatio_temporal_condition"] = ["UNetSpatioTemporalConditionModel"]
|
51 |
+
_import_structure["unets.unet_stable_cascade"] = ["StableCascadeUNet"]
|
52 |
+
_import_structure["unets.uvit_2d"] = ["UVit2DModel"]
|
53 |
+
_import_structure["vq_model"] = ["VQModel"]
|
54 |
+
|
55 |
+
if is_flax_available():
|
56 |
+
_import_structure["controlnet_flax"] = ["FlaxControlNetModel"]
|
57 |
+
_import_structure["unets.unet_2d_condition_flax"] = ["FlaxUNet2DConditionModel"]
|
58 |
+
_import_structure["vae_flax"] = ["FlaxAutoencoderKL"]
|
59 |
+
|
60 |
+
|
61 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
62 |
+
if is_torch_available():
|
63 |
+
from .adapter import MultiAdapter, T2IAdapter
|
64 |
+
from .autoencoders import (
|
65 |
+
AsymmetricAutoencoderKL,
|
66 |
+
AutoencoderKL,
|
67 |
+
AutoencoderKLTemporalDecoder,
|
68 |
+
AutoencoderTiny,
|
69 |
+
ConsistencyDecoderVAE,
|
70 |
+
)
|
71 |
+
from .controlnet import ControlNetModel
|
72 |
+
from .controlnet_xs import ControlNetXSAdapter, UNetControlNetXSModel
|
73 |
+
from .embeddings import ImageProjection
|
74 |
+
from .modeling_utils import ModelMixin
|
75 |
+
from .transformers import (
|
76 |
+
DualTransformer2DModel,
|
77 |
+
PriorTransformer,
|
78 |
+
T5FilmDecoder,
|
79 |
+
Transformer2DModel,
|
80 |
+
TransformerTemporalModel,
|
81 |
+
)
|
82 |
+
from .unets import (
|
83 |
+
I2VGenXLUNet,
|
84 |
+
Kandinsky3UNet,
|
85 |
+
MotionAdapter,
|
86 |
+
StableCascadeUNet,
|
87 |
+
UNet1DModel,
|
88 |
+
UNet2DConditionModel,
|
89 |
+
UNet2DModel,
|
90 |
+
UNet3DConditionModel,
|
91 |
+
UNetMotionModel,
|
92 |
+
UNetSpatioTemporalConditionModel,
|
93 |
+
UVit2DModel,
|
94 |
+
)
|
95 |
+
from .vq_model import VQModel
|
96 |
+
|
97 |
+
if is_flax_available():
|
98 |
+
from .controlnet_flax import FlaxControlNetModel
|
99 |
+
from .unets import FlaxUNet2DConditionModel
|
100 |
+
from .vae_flax import FlaxAutoencoderKL
|
101 |
+
|
102 |
+
else:
|
103 |
+
import sys
|
104 |
+
|
105 |
+
sys.modules[__name__] = _LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)
|
diffusers/models/autoencoders/__init__.py
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from .autoencoder_asym_kl import AsymmetricAutoencoderKL
|
2 |
+
from .autoencoder_kl import AutoencoderKL
|
3 |
+
from .autoencoder_kl_temporal_decoder import AutoencoderKLTemporalDecoder
|
4 |
+
from .autoencoder_tiny import AutoencoderTiny
|
5 |
+
from .consistency_decoder_vae import ConsistencyDecoderVAE
|
diffusers/models/transformers/__init__.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from ...utils import is_torch_available
|
2 |
+
|
3 |
+
|
4 |
+
if is_torch_available():
|
5 |
+
from .dual_transformer_2d import DualTransformer2DModel
|
6 |
+
from .prior_transformer import PriorTransformer
|
7 |
+
from .t5_film_transformer import T5FilmDecoder
|
8 |
+
from .transformer_2d import Transformer2DModel
|
9 |
+
from .transformer_temporal import TransformerTemporalModel
|
diffusers/models/unets/__init__.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from ...utils import is_flax_available, is_torch_available
|
2 |
+
|
3 |
+
|
4 |
+
if is_torch_available():
|
5 |
+
from .unet_1d import UNet1DModel
|
6 |
+
from .unet_2d import UNet2DModel
|
7 |
+
from .unet_2d_condition import UNet2DConditionModel
|
8 |
+
from .unet_3d_condition import UNet3DConditionModel
|
9 |
+
from .unet_i2vgen_xl import I2VGenXLUNet
|
10 |
+
from .unet_kandinsky3 import Kandinsky3UNet
|
11 |
+
from .unet_motion_model import MotionAdapter, UNetMotionModel
|
12 |
+
from .unet_spatio_temporal_condition import UNetSpatioTemporalConditionModel
|
13 |
+
from .unet_stable_cascade import StableCascadeUNet
|
14 |
+
from .uvit_2d import UVit2DModel
|
15 |
+
|
16 |
+
|
17 |
+
if is_flax_available():
|
18 |
+
from .unet_2d_condition_flax import FlaxUNet2DConditionModel
|
diffusers/pipelines/__init__.py
ADDED
@@ -0,0 +1,591 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ..utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_flax_available,
|
9 |
+
is_k_diffusion_available,
|
10 |
+
is_librosa_available,
|
11 |
+
is_note_seq_available,
|
12 |
+
is_onnx_available,
|
13 |
+
is_torch_available,
|
14 |
+
is_torch_npu_available,
|
15 |
+
is_transformers_available,
|
16 |
+
)
|
17 |
+
|
18 |
+
|
19 |
+
# These modules contain pipelines from multiple libraries/frameworks
|
20 |
+
_dummy_objects = {}
|
21 |
+
_import_structure = {
|
22 |
+
"controlnet": [],
|
23 |
+
"controlnet_xs": [],
|
24 |
+
"deprecated": [],
|
25 |
+
"latent_diffusion": [],
|
26 |
+
"ledits_pp": [],
|
27 |
+
"stable_diffusion": [],
|
28 |
+
"stable_diffusion_xl": [],
|
29 |
+
}
|
30 |
+
|
31 |
+
try:
|
32 |
+
if not is_torch_available():
|
33 |
+
raise OptionalDependencyNotAvailable()
|
34 |
+
except OptionalDependencyNotAvailable:
|
35 |
+
from ..utils import dummy_pt_objects # noqa F403
|
36 |
+
|
37 |
+
_dummy_objects.update(get_objects_from_module(dummy_pt_objects))
|
38 |
+
else:
|
39 |
+
_import_structure["auto_pipeline"] = [
|
40 |
+
"AutoPipelineForImage2Image",
|
41 |
+
"AutoPipelineForInpainting",
|
42 |
+
"AutoPipelineForText2Image",
|
43 |
+
]
|
44 |
+
_import_structure["consistency_models"] = ["ConsistencyModelPipeline"]
|
45 |
+
_import_structure["dance_diffusion"] = ["DanceDiffusionPipeline"]
|
46 |
+
_import_structure["ddim"] = ["DDIMPipeline"]
|
47 |
+
_import_structure["ddpm"] = ["DDPMPipeline"]
|
48 |
+
_import_structure["dit"] = ["DiTPipeline"]
|
49 |
+
_import_structure["latent_diffusion"].extend(["LDMSuperResolutionPipeline"])
|
50 |
+
_import_structure["pipeline_utils"] = [
|
51 |
+
"AudioPipelineOutput",
|
52 |
+
"DiffusionPipeline",
|
53 |
+
"StableDiffusionMixin",
|
54 |
+
"ImagePipelineOutput",
|
55 |
+
]
|
56 |
+
_import_structure["deprecated"].extend(
|
57 |
+
[
|
58 |
+
"PNDMPipeline",
|
59 |
+
"LDMPipeline",
|
60 |
+
"RePaintPipeline",
|
61 |
+
"ScoreSdeVePipeline",
|
62 |
+
"KarrasVePipeline",
|
63 |
+
]
|
64 |
+
)
|
65 |
+
try:
|
66 |
+
if not (is_torch_available() and is_librosa_available()):
|
67 |
+
raise OptionalDependencyNotAvailable()
|
68 |
+
except OptionalDependencyNotAvailable:
|
69 |
+
from ..utils import dummy_torch_and_librosa_objects # noqa F403
|
70 |
+
|
71 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_librosa_objects))
|
72 |
+
else:
|
73 |
+
_import_structure["deprecated"].extend(["AudioDiffusionPipeline", "Mel"])
|
74 |
+
|
75 |
+
try:
|
76 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
77 |
+
raise OptionalDependencyNotAvailable()
|
78 |
+
except OptionalDependencyNotAvailable:
|
79 |
+
from ..utils import dummy_transformers_and_torch_and_note_seq_objects # noqa F403
|
80 |
+
|
81 |
+
_dummy_objects.update(get_objects_from_module(dummy_transformers_and_torch_and_note_seq_objects))
|
82 |
+
else:
|
83 |
+
_import_structure["deprecated"].extend(
|
84 |
+
[
|
85 |
+
"MidiProcessor",
|
86 |
+
"SpectrogramDiffusionPipeline",
|
87 |
+
]
|
88 |
+
)
|
89 |
+
|
90 |
+
try:
|
91 |
+
if not (is_torch_available() and is_transformers_available()):
|
92 |
+
raise OptionalDependencyNotAvailable()
|
93 |
+
except OptionalDependencyNotAvailable:
|
94 |
+
from ..utils import dummy_torch_and_transformers_objects # noqa F403
|
95 |
+
|
96 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
97 |
+
else:
|
98 |
+
_import_structure["deprecated"].extend(
|
99 |
+
[
|
100 |
+
"VQDiffusionPipeline",
|
101 |
+
"AltDiffusionPipeline",
|
102 |
+
"AltDiffusionImg2ImgPipeline",
|
103 |
+
"CycleDiffusionPipeline",
|
104 |
+
"StableDiffusionInpaintPipelineLegacy",
|
105 |
+
"StableDiffusionPix2PixZeroPipeline",
|
106 |
+
"StableDiffusionParadigmsPipeline",
|
107 |
+
"StableDiffusionModelEditingPipeline",
|
108 |
+
"VersatileDiffusionDualGuidedPipeline",
|
109 |
+
"VersatileDiffusionImageVariationPipeline",
|
110 |
+
"VersatileDiffusionPipeline",
|
111 |
+
"VersatileDiffusionTextToImagePipeline",
|
112 |
+
]
|
113 |
+
)
|
114 |
+
_import_structure["amused"] = ["AmusedImg2ImgPipeline", "AmusedInpaintPipeline", "AmusedPipeline"]
|
115 |
+
_import_structure["animatediff"] = [
|
116 |
+
"AnimateDiffPipeline",
|
117 |
+
"AnimateDiffVideoToVideoPipeline",
|
118 |
+
]
|
119 |
+
_import_structure["audioldm"] = ["AudioLDMPipeline"]
|
120 |
+
_import_structure["audioldm2"] = [
|
121 |
+
"AudioLDM2Pipeline",
|
122 |
+
"AudioLDM2ProjectionModel",
|
123 |
+
"AudioLDM2UNet2DConditionModel",
|
124 |
+
]
|
125 |
+
_import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"]
|
126 |
+
_import_structure["controlnet"].extend(
|
127 |
+
[
|
128 |
+
"BlipDiffusionControlNetPipeline",
|
129 |
+
"StableDiffusionControlNetImg2ImgPipeline",
|
130 |
+
"StableDiffusionControlNetInpaintPipeline",
|
131 |
+
"StableDiffusionControlNetPipeline",
|
132 |
+
"StableDiffusionXLControlNetImg2ImgPipeline",
|
133 |
+
"StableDiffusionXLControlNetInpaintPipeline",
|
134 |
+
"StableDiffusionXLControlNetPipeline",
|
135 |
+
]
|
136 |
+
)
|
137 |
+
_import_structure["controlnet_xs"].extend(
|
138 |
+
[
|
139 |
+
"StableDiffusionControlNetXSPipeline",
|
140 |
+
"StableDiffusionXLControlNetXSPipeline",
|
141 |
+
]
|
142 |
+
)
|
143 |
+
_import_structure["deepfloyd_if"] = [
|
144 |
+
"IFImg2ImgPipeline",
|
145 |
+
"IFImg2ImgSuperResolutionPipeline",
|
146 |
+
"IFInpaintingPipeline",
|
147 |
+
"IFInpaintingSuperResolutionPipeline",
|
148 |
+
"IFPipeline",
|
149 |
+
"IFSuperResolutionPipeline",
|
150 |
+
]
|
151 |
+
_import_structure["kandinsky"] = [
|
152 |
+
"KandinskyCombinedPipeline",
|
153 |
+
"KandinskyImg2ImgCombinedPipeline",
|
154 |
+
"KandinskyImg2ImgPipeline",
|
155 |
+
"KandinskyInpaintCombinedPipeline",
|
156 |
+
"KandinskyInpaintPipeline",
|
157 |
+
"KandinskyPipeline",
|
158 |
+
"KandinskyPriorPipeline",
|
159 |
+
]
|
160 |
+
_import_structure["kandinsky2_2"] = [
|
161 |
+
"KandinskyV22CombinedPipeline",
|
162 |
+
"KandinskyV22ControlnetImg2ImgPipeline",
|
163 |
+
"KandinskyV22ControlnetPipeline",
|
164 |
+
"KandinskyV22Img2ImgCombinedPipeline",
|
165 |
+
"KandinskyV22Img2ImgPipeline",
|
166 |
+
"KandinskyV22InpaintCombinedPipeline",
|
167 |
+
"KandinskyV22InpaintPipeline",
|
168 |
+
"KandinskyV22Pipeline",
|
169 |
+
"KandinskyV22PriorEmb2EmbPipeline",
|
170 |
+
"KandinskyV22PriorPipeline",
|
171 |
+
]
|
172 |
+
_import_structure["kandinsky3"] = [
|
173 |
+
"Kandinsky3Img2ImgPipeline",
|
174 |
+
"Kandinsky3Pipeline",
|
175 |
+
]
|
176 |
+
_import_structure["latent_consistency_models"] = [
|
177 |
+
"LatentConsistencyModelImg2ImgPipeline",
|
178 |
+
"LatentConsistencyModelPipeline",
|
179 |
+
]
|
180 |
+
_import_structure["latent_diffusion"].extend(["LDMTextToImagePipeline"])
|
181 |
+
_import_structure["ledits_pp"].extend(
|
182 |
+
[
|
183 |
+
"LEditsPPPipelineStableDiffusion",
|
184 |
+
"LEditsPPPipelineStableDiffusionXL",
|
185 |
+
]
|
186 |
+
)
|
187 |
+
_import_structure["musicldm"] = ["MusicLDMPipeline"]
|
188 |
+
_import_structure["paint_by_example"] = ["PaintByExamplePipeline"]
|
189 |
+
_import_structure["pia"] = ["PIAPipeline"]
|
190 |
+
_import_structure["pixart_alpha"] = ["PixArtAlphaPipeline", "PixArtSigmaPipeline"]
|
191 |
+
_import_structure["semantic_stable_diffusion"] = ["SemanticStableDiffusionPipeline"]
|
192 |
+
_import_structure["shap_e"] = ["ShapEImg2ImgPipeline", "ShapEPipeline"]
|
193 |
+
_import_structure["stable_cascade"] = [
|
194 |
+
"StableCascadeCombinedPipeline",
|
195 |
+
"StableCascadeDecoderPipeline",
|
196 |
+
"StableCascadePriorPipeline",
|
197 |
+
]
|
198 |
+
_import_structure["stable_diffusion"].extend(
|
199 |
+
[
|
200 |
+
"CLIPImageProjection",
|
201 |
+
"StableDiffusionDepth2ImgPipeline",
|
202 |
+
"StableDiffusionImageVariationPipeline",
|
203 |
+
"StableDiffusionImg2ImgPipeline",
|
204 |
+
"StableDiffusionInpaintPipeline",
|
205 |
+
"StableDiffusionInstructPix2PixPipeline",
|
206 |
+
"StableDiffusionLatentUpscalePipeline",
|
207 |
+
"StableDiffusionPipeline",
|
208 |
+
"StableDiffusionUpscalePipeline",
|
209 |
+
"StableUnCLIPImg2ImgPipeline",
|
210 |
+
"StableUnCLIPPipeline",
|
211 |
+
"StableDiffusionLDM3DPipeline",
|
212 |
+
]
|
213 |
+
)
|
214 |
+
_import_structure["stable_diffusion_attend_and_excite"] = ["StableDiffusionAttendAndExcitePipeline"]
|
215 |
+
_import_structure["stable_diffusion_safe"] = ["StableDiffusionPipelineSafe"]
|
216 |
+
_import_structure["stable_diffusion_sag"] = ["StableDiffusionSAGPipeline"]
|
217 |
+
_import_structure["stable_diffusion_gligen"] = [
|
218 |
+
"StableDiffusionGLIGENPipeline",
|
219 |
+
"StableDiffusionGLIGENTextImagePipeline",
|
220 |
+
]
|
221 |
+
_import_structure["stable_video_diffusion"] = ["StableVideoDiffusionPipeline"]
|
222 |
+
_import_structure["stable_diffusion_xl"].extend(
|
223 |
+
[
|
224 |
+
"StableDiffusionXLImg2ImgPipeline",
|
225 |
+
"StableDiffusionXLInpaintPipeline",
|
226 |
+
"StableDiffusionXLInstructPix2PixPipeline",
|
227 |
+
"StableDiffusionXLPipeline",
|
228 |
+
]
|
229 |
+
)
|
230 |
+
_import_structure["stable_diffusion_diffedit"] = ["StableDiffusionDiffEditPipeline"]
|
231 |
+
_import_structure["stable_diffusion_ldm3d"] = ["StableDiffusionLDM3DPipeline"]
|
232 |
+
_import_structure["stable_diffusion_panorama"] = ["StableDiffusionPanoramaPipeline"]
|
233 |
+
_import_structure["t2i_adapter"] = [
|
234 |
+
"StableDiffusionAdapterPipeline",
|
235 |
+
"StableDiffusionXLAdapterPipeline",
|
236 |
+
]
|
237 |
+
_import_structure["text_to_video_synthesis"] = [
|
238 |
+
"TextToVideoSDPipeline",
|
239 |
+
"TextToVideoZeroPipeline",
|
240 |
+
"TextToVideoZeroSDXLPipeline",
|
241 |
+
"VideoToVideoSDPipeline",
|
242 |
+
]
|
243 |
+
_import_structure["i2vgen_xl"] = ["I2VGenXLPipeline"]
|
244 |
+
_import_structure["unclip"] = ["UnCLIPImageVariationPipeline", "UnCLIPPipeline"]
|
245 |
+
_import_structure["unidiffuser"] = [
|
246 |
+
"ImageTextPipelineOutput",
|
247 |
+
"UniDiffuserModel",
|
248 |
+
"UniDiffuserPipeline",
|
249 |
+
"UniDiffuserTextDecoder",
|
250 |
+
]
|
251 |
+
_import_structure["wuerstchen"] = [
|
252 |
+
"WuerstchenCombinedPipeline",
|
253 |
+
"WuerstchenDecoderPipeline",
|
254 |
+
"WuerstchenPriorPipeline",
|
255 |
+
]
|
256 |
+
try:
|
257 |
+
if not is_onnx_available():
|
258 |
+
raise OptionalDependencyNotAvailable()
|
259 |
+
except OptionalDependencyNotAvailable:
|
260 |
+
from ..utils import dummy_onnx_objects # noqa F403
|
261 |
+
|
262 |
+
_dummy_objects.update(get_objects_from_module(dummy_onnx_objects))
|
263 |
+
else:
|
264 |
+
_import_structure["onnx_utils"] = ["OnnxRuntimeModel"]
|
265 |
+
try:
|
266 |
+
if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
|
267 |
+
raise OptionalDependencyNotAvailable()
|
268 |
+
except OptionalDependencyNotAvailable:
|
269 |
+
from ..utils import dummy_torch_and_transformers_and_onnx_objects # noqa F403
|
270 |
+
|
271 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_and_onnx_objects))
|
272 |
+
else:
|
273 |
+
_import_structure["stable_diffusion"].extend(
|
274 |
+
[
|
275 |
+
"OnnxStableDiffusionImg2ImgPipeline",
|
276 |
+
"OnnxStableDiffusionInpaintPipeline",
|
277 |
+
"OnnxStableDiffusionPipeline",
|
278 |
+
"OnnxStableDiffusionUpscalePipeline",
|
279 |
+
"StableDiffusionOnnxPipeline",
|
280 |
+
]
|
281 |
+
)
|
282 |
+
|
283 |
+
try:
|
284 |
+
if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
|
285 |
+
raise OptionalDependencyNotAvailable()
|
286 |
+
except OptionalDependencyNotAvailable:
|
287 |
+
from ..utils import (
|
288 |
+
dummy_torch_and_transformers_and_k_diffusion_objects,
|
289 |
+
)
|
290 |
+
|
291 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_and_k_diffusion_objects))
|
292 |
+
else:
|
293 |
+
_import_structure["stable_diffusion_k_diffusion"] = [
|
294 |
+
"StableDiffusionKDiffusionPipeline",
|
295 |
+
"StableDiffusionXLKDiffusionPipeline",
|
296 |
+
]
|
297 |
+
try:
|
298 |
+
if not is_flax_available():
|
299 |
+
raise OptionalDependencyNotAvailable()
|
300 |
+
except OptionalDependencyNotAvailable:
|
301 |
+
from ..utils import dummy_flax_objects # noqa F403
|
302 |
+
|
303 |
+
_dummy_objects.update(get_objects_from_module(dummy_flax_objects))
|
304 |
+
else:
|
305 |
+
_import_structure["pipeline_flax_utils"] = ["FlaxDiffusionPipeline"]
|
306 |
+
try:
|
307 |
+
if not (is_flax_available() and is_transformers_available()):
|
308 |
+
raise OptionalDependencyNotAvailable()
|
309 |
+
except OptionalDependencyNotAvailable:
|
310 |
+
from ..utils import dummy_flax_and_transformers_objects # noqa F403
|
311 |
+
|
312 |
+
_dummy_objects.update(get_objects_from_module(dummy_flax_and_transformers_objects))
|
313 |
+
else:
|
314 |
+
_import_structure["controlnet"].extend(["FlaxStableDiffusionControlNetPipeline"])
|
315 |
+
_import_structure["stable_diffusion"].extend(
|
316 |
+
[
|
317 |
+
"FlaxStableDiffusionImg2ImgPipeline",
|
318 |
+
"FlaxStableDiffusionInpaintPipeline",
|
319 |
+
"FlaxStableDiffusionPipeline",
|
320 |
+
]
|
321 |
+
)
|
322 |
+
_import_structure["stable_diffusion_xl"].extend(
|
323 |
+
[
|
324 |
+
"FlaxStableDiffusionXLPipeline",
|
325 |
+
]
|
326 |
+
)
|
327 |
+
|
328 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
329 |
+
try:
|
330 |
+
if not is_torch_available():
|
331 |
+
raise OptionalDependencyNotAvailable()
|
332 |
+
except OptionalDependencyNotAvailable:
|
333 |
+
from ..utils.dummy_pt_objects import * # noqa F403
|
334 |
+
|
335 |
+
else:
|
336 |
+
from .auto_pipeline import (
|
337 |
+
AutoPipelineForImage2Image,
|
338 |
+
AutoPipelineForInpainting,
|
339 |
+
AutoPipelineForText2Image,
|
340 |
+
)
|
341 |
+
from .consistency_models import ConsistencyModelPipeline
|
342 |
+
from .dance_diffusion import DanceDiffusionPipeline
|
343 |
+
from .ddim import DDIMPipeline
|
344 |
+
from .ddpm import DDPMPipeline
|
345 |
+
from .deprecated import KarrasVePipeline, LDMPipeline, PNDMPipeline, RePaintPipeline, ScoreSdeVePipeline
|
346 |
+
from .dit import DiTPipeline
|
347 |
+
from .latent_diffusion import LDMSuperResolutionPipeline
|
348 |
+
from .pipeline_utils import (
|
349 |
+
AudioPipelineOutput,
|
350 |
+
DiffusionPipeline,
|
351 |
+
ImagePipelineOutput,
|
352 |
+
StableDiffusionMixin,
|
353 |
+
)
|
354 |
+
|
355 |
+
try:
|
356 |
+
if not (is_torch_available() and is_librosa_available()):
|
357 |
+
raise OptionalDependencyNotAvailable()
|
358 |
+
except OptionalDependencyNotAvailable:
|
359 |
+
from ..utils.dummy_torch_and_librosa_objects import *
|
360 |
+
else:
|
361 |
+
from .deprecated import AudioDiffusionPipeline, Mel
|
362 |
+
|
363 |
+
try:
|
364 |
+
if not (is_torch_available() and is_transformers_available()):
|
365 |
+
raise OptionalDependencyNotAvailable()
|
366 |
+
except OptionalDependencyNotAvailable:
|
367 |
+
from ..utils.dummy_torch_and_transformers_objects import *
|
368 |
+
else:
|
369 |
+
from .amused import AmusedImg2ImgPipeline, AmusedInpaintPipeline, AmusedPipeline
|
370 |
+
from .animatediff import AnimateDiffPipeline, AnimateDiffVideoToVideoPipeline
|
371 |
+
from .audioldm import AudioLDMPipeline
|
372 |
+
from .audioldm2 import (
|
373 |
+
AudioLDM2Pipeline,
|
374 |
+
AudioLDM2ProjectionModel,
|
375 |
+
AudioLDM2UNet2DConditionModel,
|
376 |
+
)
|
377 |
+
from .blip_diffusion import BlipDiffusionPipeline
|
378 |
+
from .controlnet import (
|
379 |
+
BlipDiffusionControlNetPipeline,
|
380 |
+
StableDiffusionControlNetImg2ImgPipeline,
|
381 |
+
StableDiffusionControlNetInpaintPipeline,
|
382 |
+
StableDiffusionControlNetPipeline,
|
383 |
+
StableDiffusionXLControlNetImg2ImgPipeline,
|
384 |
+
StableDiffusionXLControlNetInpaintPipeline,
|
385 |
+
StableDiffusionXLControlNetPipeline,
|
386 |
+
)
|
387 |
+
from .controlnet_xs import (
|
388 |
+
StableDiffusionControlNetXSPipeline,
|
389 |
+
StableDiffusionXLControlNetXSPipeline,
|
390 |
+
)
|
391 |
+
from .deepfloyd_if import (
|
392 |
+
IFImg2ImgPipeline,
|
393 |
+
IFImg2ImgSuperResolutionPipeline,
|
394 |
+
IFInpaintingPipeline,
|
395 |
+
IFInpaintingSuperResolutionPipeline,
|
396 |
+
IFPipeline,
|
397 |
+
IFSuperResolutionPipeline,
|
398 |
+
)
|
399 |
+
from .deprecated import (
|
400 |
+
AltDiffusionImg2ImgPipeline,
|
401 |
+
AltDiffusionPipeline,
|
402 |
+
CycleDiffusionPipeline,
|
403 |
+
StableDiffusionInpaintPipelineLegacy,
|
404 |
+
StableDiffusionModelEditingPipeline,
|
405 |
+
StableDiffusionParadigmsPipeline,
|
406 |
+
StableDiffusionPix2PixZeroPipeline,
|
407 |
+
VersatileDiffusionDualGuidedPipeline,
|
408 |
+
VersatileDiffusionImageVariationPipeline,
|
409 |
+
VersatileDiffusionPipeline,
|
410 |
+
VersatileDiffusionTextToImagePipeline,
|
411 |
+
VQDiffusionPipeline,
|
412 |
+
)
|
413 |
+
from .i2vgen_xl import I2VGenXLPipeline
|
414 |
+
from .kandinsky import (
|
415 |
+
KandinskyCombinedPipeline,
|
416 |
+
KandinskyImg2ImgCombinedPipeline,
|
417 |
+
KandinskyImg2ImgPipeline,
|
418 |
+
KandinskyInpaintCombinedPipeline,
|
419 |
+
KandinskyInpaintPipeline,
|
420 |
+
KandinskyPipeline,
|
421 |
+
KandinskyPriorPipeline,
|
422 |
+
)
|
423 |
+
from .kandinsky2_2 import (
|
424 |
+
KandinskyV22CombinedPipeline,
|
425 |
+
KandinskyV22ControlnetImg2ImgPipeline,
|
426 |
+
KandinskyV22ControlnetPipeline,
|
427 |
+
KandinskyV22Img2ImgCombinedPipeline,
|
428 |
+
KandinskyV22Img2ImgPipeline,
|
429 |
+
KandinskyV22InpaintCombinedPipeline,
|
430 |
+
KandinskyV22InpaintPipeline,
|
431 |
+
KandinskyV22Pipeline,
|
432 |
+
KandinskyV22PriorEmb2EmbPipeline,
|
433 |
+
KandinskyV22PriorPipeline,
|
434 |
+
)
|
435 |
+
from .kandinsky3 import (
|
436 |
+
Kandinsky3Img2ImgPipeline,
|
437 |
+
Kandinsky3Pipeline,
|
438 |
+
)
|
439 |
+
from .latent_consistency_models import (
|
440 |
+
LatentConsistencyModelImg2ImgPipeline,
|
441 |
+
LatentConsistencyModelPipeline,
|
442 |
+
)
|
443 |
+
from .latent_diffusion import LDMTextToImagePipeline
|
444 |
+
from .ledits_pp import (
|
445 |
+
LEditsPPDiffusionPipelineOutput,
|
446 |
+
LEditsPPInversionPipelineOutput,
|
447 |
+
LEditsPPPipelineStableDiffusion,
|
448 |
+
LEditsPPPipelineStableDiffusionXL,
|
449 |
+
)
|
450 |
+
from .musicldm import MusicLDMPipeline
|
451 |
+
from .paint_by_example import PaintByExamplePipeline
|
452 |
+
from .pia import PIAPipeline
|
453 |
+
from .pixart_alpha import PixArtAlphaPipeline, PixArtSigmaPipeline
|
454 |
+
from .semantic_stable_diffusion import SemanticStableDiffusionPipeline
|
455 |
+
from .shap_e import ShapEImg2ImgPipeline, ShapEPipeline
|
456 |
+
from .stable_cascade import (
|
457 |
+
StableCascadeCombinedPipeline,
|
458 |
+
StableCascadeDecoderPipeline,
|
459 |
+
StableCascadePriorPipeline,
|
460 |
+
)
|
461 |
+
from .stable_diffusion import (
|
462 |
+
CLIPImageProjection,
|
463 |
+
StableDiffusionDepth2ImgPipeline,
|
464 |
+
StableDiffusionImageVariationPipeline,
|
465 |
+
StableDiffusionImg2ImgPipeline,
|
466 |
+
StableDiffusionInpaintPipeline,
|
467 |
+
StableDiffusionInstructPix2PixPipeline,
|
468 |
+
StableDiffusionLatentUpscalePipeline,
|
469 |
+
StableDiffusionPipeline,
|
470 |
+
StableDiffusionUpscalePipeline,
|
471 |
+
StableUnCLIPImg2ImgPipeline,
|
472 |
+
StableUnCLIPPipeline,
|
473 |
+
)
|
474 |
+
from .stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline
|
475 |
+
from .stable_diffusion_diffedit import StableDiffusionDiffEditPipeline
|
476 |
+
from .stable_diffusion_gligen import StableDiffusionGLIGENPipeline, StableDiffusionGLIGENTextImagePipeline
|
477 |
+
from .stable_diffusion_ldm3d import StableDiffusionLDM3DPipeline
|
478 |
+
from .stable_diffusion_panorama import StableDiffusionPanoramaPipeline
|
479 |
+
from .stable_diffusion_safe import StableDiffusionPipelineSafe
|
480 |
+
from .stable_diffusion_sag import StableDiffusionSAGPipeline
|
481 |
+
from .stable_diffusion_xl import (
|
482 |
+
StableDiffusionXLImg2ImgPipeline,
|
483 |
+
StableDiffusionXLInpaintPipeline,
|
484 |
+
StableDiffusionXLInstructPix2PixPipeline,
|
485 |
+
StableDiffusionXLPipeline,
|
486 |
+
)
|
487 |
+
from .stable_video_diffusion import StableVideoDiffusionPipeline
|
488 |
+
from .t2i_adapter import (
|
489 |
+
StableDiffusionAdapterPipeline,
|
490 |
+
StableDiffusionXLAdapterPipeline,
|
491 |
+
)
|
492 |
+
from .text_to_video_synthesis import (
|
493 |
+
TextToVideoSDPipeline,
|
494 |
+
TextToVideoZeroPipeline,
|
495 |
+
TextToVideoZeroSDXLPipeline,
|
496 |
+
VideoToVideoSDPipeline,
|
497 |
+
)
|
498 |
+
from .unclip import UnCLIPImageVariationPipeline, UnCLIPPipeline
|
499 |
+
from .unidiffuser import (
|
500 |
+
ImageTextPipelineOutput,
|
501 |
+
UniDiffuserModel,
|
502 |
+
UniDiffuserPipeline,
|
503 |
+
UniDiffuserTextDecoder,
|
504 |
+
)
|
505 |
+
from .wuerstchen import (
|
506 |
+
WuerstchenCombinedPipeline,
|
507 |
+
WuerstchenDecoderPipeline,
|
508 |
+
WuerstchenPriorPipeline,
|
509 |
+
)
|
510 |
+
|
511 |
+
try:
|
512 |
+
if not is_onnx_available():
|
513 |
+
raise OptionalDependencyNotAvailable()
|
514 |
+
except OptionalDependencyNotAvailable:
|
515 |
+
from ..utils.dummy_onnx_objects import * # noqa F403
|
516 |
+
|
517 |
+
else:
|
518 |
+
from .onnx_utils import OnnxRuntimeModel
|
519 |
+
|
520 |
+
try:
|
521 |
+
if not (is_torch_available() and is_transformers_available() and is_onnx_available()):
|
522 |
+
raise OptionalDependencyNotAvailable()
|
523 |
+
except OptionalDependencyNotAvailable:
|
524 |
+
from ..utils.dummy_torch_and_transformers_and_onnx_objects import *
|
525 |
+
else:
|
526 |
+
from .stable_diffusion import (
|
527 |
+
OnnxStableDiffusionImg2ImgPipeline,
|
528 |
+
OnnxStableDiffusionInpaintPipeline,
|
529 |
+
OnnxStableDiffusionPipeline,
|
530 |
+
OnnxStableDiffusionUpscalePipeline,
|
531 |
+
StableDiffusionOnnxPipeline,
|
532 |
+
)
|
533 |
+
|
534 |
+
try:
|
535 |
+
if not (is_torch_available() and is_transformers_available() and is_k_diffusion_available()):
|
536 |
+
raise OptionalDependencyNotAvailable()
|
537 |
+
except OptionalDependencyNotAvailable:
|
538 |
+
from ..utils.dummy_torch_and_transformers_and_k_diffusion_objects import *
|
539 |
+
else:
|
540 |
+
from .stable_diffusion_k_diffusion import (
|
541 |
+
StableDiffusionKDiffusionPipeline,
|
542 |
+
StableDiffusionXLKDiffusionPipeline,
|
543 |
+
)
|
544 |
+
|
545 |
+
try:
|
546 |
+
if not is_flax_available():
|
547 |
+
raise OptionalDependencyNotAvailable()
|
548 |
+
except OptionalDependencyNotAvailable:
|
549 |
+
from ..utils.dummy_flax_objects import * # noqa F403
|
550 |
+
else:
|
551 |
+
from .pipeline_flax_utils import FlaxDiffusionPipeline
|
552 |
+
|
553 |
+
try:
|
554 |
+
if not (is_flax_available() and is_transformers_available()):
|
555 |
+
raise OptionalDependencyNotAvailable()
|
556 |
+
except OptionalDependencyNotAvailable:
|
557 |
+
from ..utils.dummy_flax_and_transformers_objects import *
|
558 |
+
else:
|
559 |
+
from .controlnet import FlaxStableDiffusionControlNetPipeline
|
560 |
+
from .stable_diffusion import (
|
561 |
+
FlaxStableDiffusionImg2ImgPipeline,
|
562 |
+
FlaxStableDiffusionInpaintPipeline,
|
563 |
+
FlaxStableDiffusionPipeline,
|
564 |
+
)
|
565 |
+
from .stable_diffusion_xl import (
|
566 |
+
FlaxStableDiffusionXLPipeline,
|
567 |
+
)
|
568 |
+
|
569 |
+
try:
|
570 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
571 |
+
raise OptionalDependencyNotAvailable()
|
572 |
+
except OptionalDependencyNotAvailable:
|
573 |
+
from ..utils.dummy_transformers_and_torch_and_note_seq_objects import * # noqa F403
|
574 |
+
|
575 |
+
else:
|
576 |
+
from .deprecated import (
|
577 |
+
MidiProcessor,
|
578 |
+
SpectrogramDiffusionPipeline,
|
579 |
+
)
|
580 |
+
|
581 |
+
else:
|
582 |
+
import sys
|
583 |
+
|
584 |
+
sys.modules[__name__] = _LazyModule(
|
585 |
+
__name__,
|
586 |
+
globals()["__file__"],
|
587 |
+
_import_structure,
|
588 |
+
module_spec=__spec__,
|
589 |
+
)
|
590 |
+
for name, value in _dummy_objects.items():
|
591 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/amused/__init__.py
ADDED
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
is_torch_available,
|
8 |
+
is_transformers_available,
|
9 |
+
)
|
10 |
+
|
11 |
+
|
12 |
+
_dummy_objects = {}
|
13 |
+
_import_structure = {}
|
14 |
+
|
15 |
+
try:
|
16 |
+
if not (is_transformers_available() and is_torch_available()):
|
17 |
+
raise OptionalDependencyNotAvailable()
|
18 |
+
except OptionalDependencyNotAvailable:
|
19 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
20 |
+
AmusedImg2ImgPipeline,
|
21 |
+
AmusedInpaintPipeline,
|
22 |
+
AmusedPipeline,
|
23 |
+
)
|
24 |
+
|
25 |
+
_dummy_objects.update(
|
26 |
+
{
|
27 |
+
"AmusedPipeline": AmusedPipeline,
|
28 |
+
"AmusedImg2ImgPipeline": AmusedImg2ImgPipeline,
|
29 |
+
"AmusedInpaintPipeline": AmusedInpaintPipeline,
|
30 |
+
}
|
31 |
+
)
|
32 |
+
else:
|
33 |
+
_import_structure["pipeline_amused"] = ["AmusedPipeline"]
|
34 |
+
_import_structure["pipeline_amused_img2img"] = ["AmusedImg2ImgPipeline"]
|
35 |
+
_import_structure["pipeline_amused_inpaint"] = ["AmusedInpaintPipeline"]
|
36 |
+
|
37 |
+
|
38 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
39 |
+
try:
|
40 |
+
if not (is_transformers_available() and is_torch_available()):
|
41 |
+
raise OptionalDependencyNotAvailable()
|
42 |
+
except OptionalDependencyNotAvailable:
|
43 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
44 |
+
AmusedPipeline,
|
45 |
+
)
|
46 |
+
else:
|
47 |
+
from .pipeline_amused import AmusedPipeline
|
48 |
+
from .pipeline_amused_img2img import AmusedImg2ImgPipeline
|
49 |
+
from .pipeline_amused_inpaint import AmusedInpaintPipeline
|
50 |
+
|
51 |
+
else:
|
52 |
+
import sys
|
53 |
+
|
54 |
+
sys.modules[__name__] = _LazyModule(
|
55 |
+
__name__,
|
56 |
+
globals()["__file__"],
|
57 |
+
_import_structure,
|
58 |
+
module_spec=__spec__,
|
59 |
+
)
|
60 |
+
|
61 |
+
for name, value in _dummy_objects.items():
|
62 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/animatediff/__init__.py
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {"pipeline_output": ["AnimateDiffPipelineOutput"]}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_animatediff"] = ["AnimateDiffPipeline"]
|
25 |
+
_import_structure["pipeline_animatediff_video2video"] = ["AnimateDiffVideoToVideoPipeline"]
|
26 |
+
|
27 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
28 |
+
try:
|
29 |
+
if not (is_transformers_available() and is_torch_available()):
|
30 |
+
raise OptionalDependencyNotAvailable()
|
31 |
+
except OptionalDependencyNotAvailable:
|
32 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
33 |
+
|
34 |
+
else:
|
35 |
+
from .pipeline_animatediff import AnimateDiffPipeline
|
36 |
+
from .pipeline_animatediff_video2video import AnimateDiffVideoToVideoPipeline
|
37 |
+
from .pipeline_output import AnimateDiffPipelineOutput
|
38 |
+
|
39 |
+
else:
|
40 |
+
import sys
|
41 |
+
|
42 |
+
sys.modules[__name__] = _LazyModule(
|
43 |
+
__name__,
|
44 |
+
globals()["__file__"],
|
45 |
+
_import_structure,
|
46 |
+
module_spec=__spec__,
|
47 |
+
)
|
48 |
+
for name, value in _dummy_objects.items():
|
49 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/audioldm/__init__.py
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
is_torch_available,
|
8 |
+
is_transformers_available,
|
9 |
+
is_transformers_version,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
21 |
+
AudioLDMPipeline,
|
22 |
+
)
|
23 |
+
|
24 |
+
_dummy_objects.update({"AudioLDMPipeline": AudioLDMPipeline})
|
25 |
+
else:
|
26 |
+
_import_structure["pipeline_audioldm"] = ["AudioLDMPipeline"]
|
27 |
+
|
28 |
+
|
29 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
30 |
+
try:
|
31 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
32 |
+
raise OptionalDependencyNotAvailable()
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
35 |
+
AudioLDMPipeline,
|
36 |
+
)
|
37 |
+
|
38 |
+
else:
|
39 |
+
from .pipeline_audioldm import AudioLDMPipeline
|
40 |
+
else:
|
41 |
+
import sys
|
42 |
+
|
43 |
+
sys.modules[__name__] = _LazyModule(
|
44 |
+
__name__,
|
45 |
+
globals()["__file__"],
|
46 |
+
_import_structure,
|
47 |
+
module_spec=__spec__,
|
48 |
+
)
|
49 |
+
|
50 |
+
for name, value in _dummy_objects.items():
|
51 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/audioldm2/__init__.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
is_transformers_version,
|
11 |
+
)
|
12 |
+
|
13 |
+
|
14 |
+
_dummy_objects = {}
|
15 |
+
_import_structure = {}
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["modeling_audioldm2"] = ["AudioLDM2ProjectionModel", "AudioLDM2UNet2DConditionModel"]
|
26 |
+
_import_structure["pipeline_audioldm2"] = ["AudioLDM2Pipeline"]
|
27 |
+
|
28 |
+
|
29 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
30 |
+
try:
|
31 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
32 |
+
raise OptionalDependencyNotAvailable()
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
|
36 |
+
else:
|
37 |
+
from .modeling_audioldm2 import AudioLDM2ProjectionModel, AudioLDM2UNet2DConditionModel
|
38 |
+
from .pipeline_audioldm2 import AudioLDM2Pipeline
|
39 |
+
|
40 |
+
else:
|
41 |
+
import sys
|
42 |
+
|
43 |
+
sys.modules[__name__] = _LazyModule(
|
44 |
+
__name__,
|
45 |
+
globals()["__file__"],
|
46 |
+
_import_structure,
|
47 |
+
module_spec=__spec__,
|
48 |
+
)
|
49 |
+
for name, value in _dummy_objects.items():
|
50 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/blip_diffusion/__init__.py
ADDED
@@ -0,0 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from dataclasses import dataclass
|
2 |
+
from typing import List, Optional, Union
|
3 |
+
|
4 |
+
import numpy as np
|
5 |
+
import PIL
|
6 |
+
from PIL import Image
|
7 |
+
|
8 |
+
from ...utils import OptionalDependencyNotAvailable, is_torch_available, is_transformers_available
|
9 |
+
|
10 |
+
|
11 |
+
try:
|
12 |
+
if not (is_transformers_available() and is_torch_available()):
|
13 |
+
raise OptionalDependencyNotAvailable()
|
14 |
+
except OptionalDependencyNotAvailable:
|
15 |
+
from ...utils.dummy_torch_and_transformers_objects import ShapEPipeline
|
16 |
+
else:
|
17 |
+
from .blip_image_processing import BlipImageProcessor
|
18 |
+
from .modeling_blip2 import Blip2QFormerModel
|
19 |
+
from .modeling_ctx_clip import ContextCLIPTextModel
|
20 |
+
from .pipeline_blip_diffusion import BlipDiffusionPipeline
|
diffusers/pipelines/consistency_models/__init__.py
ADDED
@@ -0,0 +1,24 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
_LazyModule,
|
6 |
+
)
|
7 |
+
|
8 |
+
|
9 |
+
_import_structure = {
|
10 |
+
"pipeline_consistency_models": ["ConsistencyModelPipeline"],
|
11 |
+
}
|
12 |
+
|
13 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
14 |
+
from .pipeline_consistency_models import ConsistencyModelPipeline
|
15 |
+
|
16 |
+
else:
|
17 |
+
import sys
|
18 |
+
|
19 |
+
sys.modules[__name__] = _LazyModule(
|
20 |
+
__name__,
|
21 |
+
globals()["__file__"],
|
22 |
+
_import_structure,
|
23 |
+
module_spec=__spec__,
|
24 |
+
)
|
diffusers/pipelines/controlnet/__init__.py
ADDED
@@ -0,0 +1,80 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_flax_available,
|
9 |
+
is_torch_available,
|
10 |
+
is_transformers_available,
|
11 |
+
)
|
12 |
+
|
13 |
+
|
14 |
+
_dummy_objects = {}
|
15 |
+
_import_structure = {}
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available()):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["multicontrolnet"] = ["MultiControlNetModel"]
|
26 |
+
_import_structure["pipeline_controlnet"] = ["StableDiffusionControlNetPipeline"]
|
27 |
+
_import_structure["pipeline_controlnet_blip_diffusion"] = ["BlipDiffusionControlNetPipeline"]
|
28 |
+
_import_structure["pipeline_controlnet_img2img"] = ["StableDiffusionControlNetImg2ImgPipeline"]
|
29 |
+
_import_structure["pipeline_controlnet_inpaint"] = ["StableDiffusionControlNetInpaintPipeline"]
|
30 |
+
_import_structure["pipeline_controlnet_inpaint_sd_xl"] = ["StableDiffusionXLControlNetInpaintPipeline"]
|
31 |
+
_import_structure["pipeline_controlnet_sd_xl"] = ["StableDiffusionXLControlNetPipeline"]
|
32 |
+
_import_structure["pipeline_controlnet_sd_xl_img2img"] = ["StableDiffusionXLControlNetImg2ImgPipeline"]
|
33 |
+
try:
|
34 |
+
if not (is_transformers_available() and is_flax_available()):
|
35 |
+
raise OptionalDependencyNotAvailable()
|
36 |
+
except OptionalDependencyNotAvailable:
|
37 |
+
from ...utils import dummy_flax_and_transformers_objects # noqa F403
|
38 |
+
|
39 |
+
_dummy_objects.update(get_objects_from_module(dummy_flax_and_transformers_objects))
|
40 |
+
else:
|
41 |
+
_import_structure["pipeline_flax_controlnet"] = ["FlaxStableDiffusionControlNetPipeline"]
|
42 |
+
|
43 |
+
|
44 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
45 |
+
try:
|
46 |
+
if not (is_transformers_available() and is_torch_available()):
|
47 |
+
raise OptionalDependencyNotAvailable()
|
48 |
+
|
49 |
+
except OptionalDependencyNotAvailable:
|
50 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
51 |
+
else:
|
52 |
+
from .multicontrolnet import MultiControlNetModel
|
53 |
+
from .pipeline_controlnet import StableDiffusionControlNetPipeline
|
54 |
+
from .pipeline_controlnet_blip_diffusion import BlipDiffusionControlNetPipeline
|
55 |
+
from .pipeline_controlnet_img2img import StableDiffusionControlNetImg2ImgPipeline
|
56 |
+
from .pipeline_controlnet_inpaint import StableDiffusionControlNetInpaintPipeline
|
57 |
+
from .pipeline_controlnet_inpaint_sd_xl import StableDiffusionXLControlNetInpaintPipeline
|
58 |
+
from .pipeline_controlnet_sd_xl import StableDiffusionXLControlNetPipeline
|
59 |
+
from .pipeline_controlnet_sd_xl_img2img import StableDiffusionXLControlNetImg2ImgPipeline
|
60 |
+
|
61 |
+
try:
|
62 |
+
if not (is_transformers_available() and is_flax_available()):
|
63 |
+
raise OptionalDependencyNotAvailable()
|
64 |
+
except OptionalDependencyNotAvailable:
|
65 |
+
from ...utils.dummy_flax_and_transformers_objects import * # noqa F403
|
66 |
+
else:
|
67 |
+
from .pipeline_flax_controlnet import FlaxStableDiffusionControlNetPipeline
|
68 |
+
|
69 |
+
|
70 |
+
else:
|
71 |
+
import sys
|
72 |
+
|
73 |
+
sys.modules[__name__] = _LazyModule(
|
74 |
+
__name__,
|
75 |
+
globals()["__file__"],
|
76 |
+
_import_structure,
|
77 |
+
module_spec=__spec__,
|
78 |
+
)
|
79 |
+
for name, value in _dummy_objects.items():
|
80 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/controlnet_xs/__init__.py
ADDED
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_flax_available,
|
9 |
+
is_torch_available,
|
10 |
+
is_transformers_available,
|
11 |
+
)
|
12 |
+
|
13 |
+
|
14 |
+
_dummy_objects = {}
|
15 |
+
_import_structure = {}
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available()):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["pipeline_controlnet_xs"] = ["StableDiffusionControlNetXSPipeline"]
|
26 |
+
_import_structure["pipeline_controlnet_xs_sd_xl"] = ["StableDiffusionXLControlNetXSPipeline"]
|
27 |
+
try:
|
28 |
+
if not (is_transformers_available() and is_flax_available()):
|
29 |
+
raise OptionalDependencyNotAvailable()
|
30 |
+
except OptionalDependencyNotAvailable:
|
31 |
+
from ...utils import dummy_flax_and_transformers_objects # noqa F403
|
32 |
+
|
33 |
+
_dummy_objects.update(get_objects_from_module(dummy_flax_and_transformers_objects))
|
34 |
+
else:
|
35 |
+
pass # _import_structure["pipeline_flax_controlnet"] = ["FlaxStableDiffusionControlNetPipeline"]
|
36 |
+
|
37 |
+
|
38 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
39 |
+
try:
|
40 |
+
if not (is_transformers_available() and is_torch_available()):
|
41 |
+
raise OptionalDependencyNotAvailable()
|
42 |
+
|
43 |
+
except OptionalDependencyNotAvailable:
|
44 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
45 |
+
else:
|
46 |
+
from .pipeline_controlnet_xs import StableDiffusionControlNetXSPipeline
|
47 |
+
from .pipeline_controlnet_xs_sd_xl import StableDiffusionXLControlNetXSPipeline
|
48 |
+
|
49 |
+
try:
|
50 |
+
if not (is_transformers_available() and is_flax_available()):
|
51 |
+
raise OptionalDependencyNotAvailable()
|
52 |
+
except OptionalDependencyNotAvailable:
|
53 |
+
from ...utils.dummy_flax_and_transformers_objects import * # noqa F403
|
54 |
+
else:
|
55 |
+
pass # from .pipeline_flax_controlnet import FlaxStableDiffusionControlNetPipeline
|
56 |
+
|
57 |
+
|
58 |
+
else:
|
59 |
+
import sys
|
60 |
+
|
61 |
+
sys.modules[__name__] = _LazyModule(
|
62 |
+
__name__,
|
63 |
+
globals()["__file__"],
|
64 |
+
_import_structure,
|
65 |
+
module_spec=__spec__,
|
66 |
+
)
|
67 |
+
for name, value in _dummy_objects.items():
|
68 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/dance_diffusion/__init__.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_dance_diffusion": ["DanceDiffusionPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_dance_diffusion import DanceDiffusionPipeline
|
10 |
+
else:
|
11 |
+
import sys
|
12 |
+
|
13 |
+
sys.modules[__name__] = _LazyModule(
|
14 |
+
__name__,
|
15 |
+
globals()["__file__"],
|
16 |
+
_import_structure,
|
17 |
+
module_spec=__spec__,
|
18 |
+
)
|
diffusers/pipelines/ddim/__init__.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_ddim": ["DDIMPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_ddim import DDIMPipeline
|
10 |
+
else:
|
11 |
+
import sys
|
12 |
+
|
13 |
+
sys.modules[__name__] = _LazyModule(
|
14 |
+
__name__,
|
15 |
+
globals()["__file__"],
|
16 |
+
_import_structure,
|
17 |
+
module_spec=__spec__,
|
18 |
+
)
|
diffusers/pipelines/ddpm/__init__.py
ADDED
@@ -0,0 +1,22 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
_LazyModule,
|
6 |
+
)
|
7 |
+
|
8 |
+
|
9 |
+
_import_structure = {"pipeline_ddpm": ["DDPMPipeline"]}
|
10 |
+
|
11 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
12 |
+
from .pipeline_ddpm import DDPMPipeline
|
13 |
+
|
14 |
+
else:
|
15 |
+
import sys
|
16 |
+
|
17 |
+
sys.modules[__name__] = _LazyModule(
|
18 |
+
__name__,
|
19 |
+
globals()["__file__"],
|
20 |
+
_import_structure,
|
21 |
+
module_spec=__spec__,
|
22 |
+
)
|
diffusers/pipelines/deepfloyd_if/__init__.py
ADDED
@@ -0,0 +1,85 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {
|
15 |
+
"timesteps": [
|
16 |
+
"fast27_timesteps",
|
17 |
+
"smart100_timesteps",
|
18 |
+
"smart185_timesteps",
|
19 |
+
"smart27_timesteps",
|
20 |
+
"smart50_timesteps",
|
21 |
+
"super100_timesteps",
|
22 |
+
"super27_timesteps",
|
23 |
+
"super40_timesteps",
|
24 |
+
]
|
25 |
+
}
|
26 |
+
|
27 |
+
try:
|
28 |
+
if not (is_transformers_available() and is_torch_available()):
|
29 |
+
raise OptionalDependencyNotAvailable()
|
30 |
+
except OptionalDependencyNotAvailable:
|
31 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
32 |
+
|
33 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
34 |
+
else:
|
35 |
+
_import_structure["pipeline_if"] = ["IFPipeline"]
|
36 |
+
_import_structure["pipeline_if_img2img"] = ["IFImg2ImgPipeline"]
|
37 |
+
_import_structure["pipeline_if_img2img_superresolution"] = ["IFImg2ImgSuperResolutionPipeline"]
|
38 |
+
_import_structure["pipeline_if_inpainting"] = ["IFInpaintingPipeline"]
|
39 |
+
_import_structure["pipeline_if_inpainting_superresolution"] = ["IFInpaintingSuperResolutionPipeline"]
|
40 |
+
_import_structure["pipeline_if_superresolution"] = ["IFSuperResolutionPipeline"]
|
41 |
+
_import_structure["pipeline_output"] = ["IFPipelineOutput"]
|
42 |
+
_import_structure["safety_checker"] = ["IFSafetyChecker"]
|
43 |
+
_import_structure["watermark"] = ["IFWatermarker"]
|
44 |
+
|
45 |
+
|
46 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
47 |
+
try:
|
48 |
+
if not (is_transformers_available() and is_torch_available()):
|
49 |
+
raise OptionalDependencyNotAvailable()
|
50 |
+
|
51 |
+
except OptionalDependencyNotAvailable:
|
52 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
53 |
+
else:
|
54 |
+
from .pipeline_if import IFPipeline
|
55 |
+
from .pipeline_if_img2img import IFImg2ImgPipeline
|
56 |
+
from .pipeline_if_img2img_superresolution import IFImg2ImgSuperResolutionPipeline
|
57 |
+
from .pipeline_if_inpainting import IFInpaintingPipeline
|
58 |
+
from .pipeline_if_inpainting_superresolution import IFInpaintingSuperResolutionPipeline
|
59 |
+
from .pipeline_if_superresolution import IFSuperResolutionPipeline
|
60 |
+
from .pipeline_output import IFPipelineOutput
|
61 |
+
from .safety_checker import IFSafetyChecker
|
62 |
+
from .timesteps import (
|
63 |
+
fast27_timesteps,
|
64 |
+
smart27_timesteps,
|
65 |
+
smart50_timesteps,
|
66 |
+
smart100_timesteps,
|
67 |
+
smart185_timesteps,
|
68 |
+
super27_timesteps,
|
69 |
+
super40_timesteps,
|
70 |
+
super100_timesteps,
|
71 |
+
)
|
72 |
+
from .watermark import IFWatermarker
|
73 |
+
|
74 |
+
else:
|
75 |
+
import sys
|
76 |
+
|
77 |
+
sys.modules[__name__] = _LazyModule(
|
78 |
+
__name__,
|
79 |
+
globals()["__file__"],
|
80 |
+
_import_structure,
|
81 |
+
module_spec=__spec__,
|
82 |
+
)
|
83 |
+
|
84 |
+
for name, value in _dummy_objects.items():
|
85 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/__init__.py
ADDED
@@ -0,0 +1,153 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_librosa_available,
|
9 |
+
is_note_seq_available,
|
10 |
+
is_torch_available,
|
11 |
+
is_transformers_available,
|
12 |
+
)
|
13 |
+
|
14 |
+
|
15 |
+
_dummy_objects = {}
|
16 |
+
_import_structure = {}
|
17 |
+
|
18 |
+
try:
|
19 |
+
if not is_torch_available():
|
20 |
+
raise OptionalDependencyNotAvailable()
|
21 |
+
except OptionalDependencyNotAvailable:
|
22 |
+
from ...utils import dummy_pt_objects
|
23 |
+
|
24 |
+
_dummy_objects.update(get_objects_from_module(dummy_pt_objects))
|
25 |
+
else:
|
26 |
+
_import_structure["latent_diffusion_uncond"] = ["LDMPipeline"]
|
27 |
+
_import_structure["pndm"] = ["PNDMPipeline"]
|
28 |
+
_import_structure["repaint"] = ["RePaintPipeline"]
|
29 |
+
_import_structure["score_sde_ve"] = ["ScoreSdeVePipeline"]
|
30 |
+
_import_structure["stochastic_karras_ve"] = ["KarrasVePipeline"]
|
31 |
+
|
32 |
+
try:
|
33 |
+
if not (is_transformers_available() and is_torch_available()):
|
34 |
+
raise OptionalDependencyNotAvailable()
|
35 |
+
except OptionalDependencyNotAvailable:
|
36 |
+
from ...utils import dummy_torch_and_transformers_objects
|
37 |
+
|
38 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
39 |
+
else:
|
40 |
+
_import_structure["alt_diffusion"] = [
|
41 |
+
"AltDiffusionImg2ImgPipeline",
|
42 |
+
"AltDiffusionPipeline",
|
43 |
+
"AltDiffusionPipelineOutput",
|
44 |
+
]
|
45 |
+
_import_structure["versatile_diffusion"] = [
|
46 |
+
"VersatileDiffusionDualGuidedPipeline",
|
47 |
+
"VersatileDiffusionImageVariationPipeline",
|
48 |
+
"VersatileDiffusionPipeline",
|
49 |
+
"VersatileDiffusionTextToImagePipeline",
|
50 |
+
]
|
51 |
+
_import_structure["vq_diffusion"] = ["VQDiffusionPipeline"]
|
52 |
+
_import_structure["stable_diffusion_variants"] = [
|
53 |
+
"CycleDiffusionPipeline",
|
54 |
+
"StableDiffusionInpaintPipelineLegacy",
|
55 |
+
"StableDiffusionPix2PixZeroPipeline",
|
56 |
+
"StableDiffusionParadigmsPipeline",
|
57 |
+
"StableDiffusionModelEditingPipeline",
|
58 |
+
]
|
59 |
+
|
60 |
+
try:
|
61 |
+
if not (is_torch_available() and is_librosa_available()):
|
62 |
+
raise OptionalDependencyNotAvailable()
|
63 |
+
except OptionalDependencyNotAvailable:
|
64 |
+
from ...utils import dummy_torch_and_librosa_objects # noqa F403
|
65 |
+
|
66 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_librosa_objects))
|
67 |
+
|
68 |
+
else:
|
69 |
+
_import_structure["audio_diffusion"] = ["AudioDiffusionPipeline", "Mel"]
|
70 |
+
|
71 |
+
try:
|
72 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
73 |
+
raise OptionalDependencyNotAvailable()
|
74 |
+
except OptionalDependencyNotAvailable:
|
75 |
+
from ...utils import dummy_transformers_and_torch_and_note_seq_objects # noqa F403
|
76 |
+
|
77 |
+
_dummy_objects.update(get_objects_from_module(dummy_transformers_and_torch_and_note_seq_objects))
|
78 |
+
|
79 |
+
else:
|
80 |
+
_import_structure["spectrogram_diffusion"] = ["MidiProcessor", "SpectrogramDiffusionPipeline"]
|
81 |
+
|
82 |
+
|
83 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
84 |
+
try:
|
85 |
+
if not is_torch_available():
|
86 |
+
raise OptionalDependencyNotAvailable()
|
87 |
+
except OptionalDependencyNotAvailable:
|
88 |
+
from ...utils.dummy_pt_objects import *
|
89 |
+
|
90 |
+
else:
|
91 |
+
from .latent_diffusion_uncond import LDMPipeline
|
92 |
+
from .pndm import PNDMPipeline
|
93 |
+
from .repaint import RePaintPipeline
|
94 |
+
from .score_sde_ve import ScoreSdeVePipeline
|
95 |
+
from .stochastic_karras_ve import KarrasVePipeline
|
96 |
+
|
97 |
+
try:
|
98 |
+
if not (is_transformers_available() and is_torch_available()):
|
99 |
+
raise OptionalDependencyNotAvailable()
|
100 |
+
except OptionalDependencyNotAvailable:
|
101 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
102 |
+
|
103 |
+
else:
|
104 |
+
from .alt_diffusion import AltDiffusionImg2ImgPipeline, AltDiffusionPipeline, AltDiffusionPipelineOutput
|
105 |
+
from .audio_diffusion import AudioDiffusionPipeline, Mel
|
106 |
+
from .spectrogram_diffusion import SpectrogramDiffusionPipeline
|
107 |
+
from .stable_diffusion_variants import (
|
108 |
+
CycleDiffusionPipeline,
|
109 |
+
StableDiffusionInpaintPipelineLegacy,
|
110 |
+
StableDiffusionModelEditingPipeline,
|
111 |
+
StableDiffusionParadigmsPipeline,
|
112 |
+
StableDiffusionPix2PixZeroPipeline,
|
113 |
+
)
|
114 |
+
from .stochastic_karras_ve import KarrasVePipeline
|
115 |
+
from .versatile_diffusion import (
|
116 |
+
VersatileDiffusionDualGuidedPipeline,
|
117 |
+
VersatileDiffusionImageVariationPipeline,
|
118 |
+
VersatileDiffusionPipeline,
|
119 |
+
VersatileDiffusionTextToImagePipeline,
|
120 |
+
)
|
121 |
+
from .vq_diffusion import VQDiffusionPipeline
|
122 |
+
|
123 |
+
try:
|
124 |
+
if not (is_torch_available() and is_librosa_available()):
|
125 |
+
raise OptionalDependencyNotAvailable()
|
126 |
+
except OptionalDependencyNotAvailable:
|
127 |
+
from ...utils.dummy_torch_and_librosa_objects import *
|
128 |
+
else:
|
129 |
+
from .audio_diffusion import AudioDiffusionPipeline, Mel
|
130 |
+
|
131 |
+
try:
|
132 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
133 |
+
raise OptionalDependencyNotAvailable()
|
134 |
+
except OptionalDependencyNotAvailable:
|
135 |
+
from ...utils.dummy_transformers_and_torch_and_note_seq_objects import * # noqa F403
|
136 |
+
else:
|
137 |
+
from .spectrogram_diffusion import (
|
138 |
+
MidiProcessor,
|
139 |
+
SpectrogramDiffusionPipeline,
|
140 |
+
)
|
141 |
+
|
142 |
+
|
143 |
+
else:
|
144 |
+
import sys
|
145 |
+
|
146 |
+
sys.modules[__name__] = _LazyModule(
|
147 |
+
__name__,
|
148 |
+
globals()["__file__"],
|
149 |
+
_import_structure,
|
150 |
+
module_spec=__spec__,
|
151 |
+
)
|
152 |
+
for name, value in _dummy_objects.items():
|
153 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/alt_diffusion/__init__.py
ADDED
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ....utils import dummy_torch_and_transformers_objects
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["modeling_roberta_series"] = ["RobertaSeriesModelWithTransformation"]
|
25 |
+
_import_structure["pipeline_alt_diffusion"] = ["AltDiffusionPipeline"]
|
26 |
+
_import_structure["pipeline_alt_diffusion_img2img"] = ["AltDiffusionImg2ImgPipeline"]
|
27 |
+
|
28 |
+
_import_structure["pipeline_output"] = ["AltDiffusionPipelineOutput"]
|
29 |
+
|
30 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
31 |
+
try:
|
32 |
+
if not (is_transformers_available() and is_torch_available()):
|
33 |
+
raise OptionalDependencyNotAvailable()
|
34 |
+
except OptionalDependencyNotAvailable:
|
35 |
+
from ....utils.dummy_torch_and_transformers_objects import *
|
36 |
+
|
37 |
+
else:
|
38 |
+
from .modeling_roberta_series import RobertaSeriesModelWithTransformation
|
39 |
+
from .pipeline_alt_diffusion import AltDiffusionPipeline
|
40 |
+
from .pipeline_alt_diffusion_img2img import AltDiffusionImg2ImgPipeline
|
41 |
+
from .pipeline_output import AltDiffusionPipelineOutput
|
42 |
+
|
43 |
+
else:
|
44 |
+
import sys
|
45 |
+
|
46 |
+
sys.modules[__name__] = _LazyModule(
|
47 |
+
__name__,
|
48 |
+
globals()["__file__"],
|
49 |
+
_import_structure,
|
50 |
+
module_spec=__spec__,
|
51 |
+
)
|
52 |
+
for name, value in _dummy_objects.items():
|
53 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/audio_diffusion/__init__.py
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {
|
7 |
+
"mel": ["Mel"],
|
8 |
+
"pipeline_audio_diffusion": ["AudioDiffusionPipeline"],
|
9 |
+
}
|
10 |
+
|
11 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
12 |
+
from .mel import Mel
|
13 |
+
from .pipeline_audio_diffusion import AudioDiffusionPipeline
|
14 |
+
|
15 |
+
else:
|
16 |
+
import sys
|
17 |
+
|
18 |
+
sys.modules[__name__] = _LazyModule(
|
19 |
+
__name__,
|
20 |
+
globals()["__file__"],
|
21 |
+
_import_structure,
|
22 |
+
module_spec=__spec__,
|
23 |
+
)
|
diffusers/pipelines/deprecated/latent_diffusion_uncond/__init__.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_latent_diffusion_uncond": ["LDMPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_latent_diffusion_uncond import LDMPipeline
|
10 |
+
else:
|
11 |
+
import sys
|
12 |
+
|
13 |
+
sys.modules[__name__] = _LazyModule(
|
14 |
+
__name__,
|
15 |
+
globals()["__file__"],
|
16 |
+
_import_structure,
|
17 |
+
module_spec=__spec__,
|
18 |
+
)
|
diffusers/pipelines/deprecated/pndm/__init__.py
ADDED
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_pndm": ["PNDMPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_pndm import PNDMPipeline
|
10 |
+
else:
|
11 |
+
import sys
|
12 |
+
|
13 |
+
sys.modules[__name__] = _LazyModule(
|
14 |
+
__name__,
|
15 |
+
globals()["__file__"],
|
16 |
+
_import_structure,
|
17 |
+
module_spec=__spec__,
|
18 |
+
)
|
diffusers/pipelines/deprecated/repaint/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_repaint": ["RePaintPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_repaint import RePaintPipeline
|
10 |
+
|
11 |
+
else:
|
12 |
+
import sys
|
13 |
+
|
14 |
+
sys.modules[__name__] = _LazyModule(
|
15 |
+
__name__,
|
16 |
+
globals()["__file__"],
|
17 |
+
_import_structure,
|
18 |
+
module_spec=__spec__,
|
19 |
+
)
|
diffusers/pipelines/deprecated/score_sde_ve/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_score_sde_ve": ["ScoreSdeVePipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_score_sde_ve import ScoreSdeVePipeline
|
10 |
+
|
11 |
+
else:
|
12 |
+
import sys
|
13 |
+
|
14 |
+
sys.modules[__name__] = _LazyModule(
|
15 |
+
__name__,
|
16 |
+
globals()["__file__"],
|
17 |
+
_import_structure,
|
18 |
+
module_spec=__spec__,
|
19 |
+
)
|
diffusers/pipelines/deprecated/spectrogram_diffusion/__init__.py
ADDED
@@ -0,0 +1,75 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# flake8: noqa
|
2 |
+
from typing import TYPE_CHECKING
|
3 |
+
from ....utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
_LazyModule,
|
6 |
+
is_note_seq_available,
|
7 |
+
OptionalDependencyNotAvailable,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
get_objects_from_module,
|
11 |
+
)
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ....utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["continous_encoder"] = ["SpectrogramContEncoder"]
|
25 |
+
_import_structure["notes_encoder"] = ["SpectrogramNotesEncoder"]
|
26 |
+
_import_structure["pipeline_spectrogram_diffusion"] = [
|
27 |
+
"SpectrogramContEncoder",
|
28 |
+
"SpectrogramDiffusionPipeline",
|
29 |
+
"T5FilmDecoder",
|
30 |
+
]
|
31 |
+
try:
|
32 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
33 |
+
raise OptionalDependencyNotAvailable()
|
34 |
+
except OptionalDependencyNotAvailable:
|
35 |
+
from ....utils import dummy_transformers_and_torch_and_note_seq_objects
|
36 |
+
|
37 |
+
_dummy_objects.update(get_objects_from_module(dummy_transformers_and_torch_and_note_seq_objects))
|
38 |
+
else:
|
39 |
+
_import_structure["midi_utils"] = ["MidiProcessor"]
|
40 |
+
|
41 |
+
|
42 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
43 |
+
try:
|
44 |
+
if not (is_transformers_available() and is_torch_available()):
|
45 |
+
raise OptionalDependencyNotAvailable()
|
46 |
+
|
47 |
+
except OptionalDependencyNotAvailable:
|
48 |
+
from ....utils.dummy_torch_and_transformers_objects import *
|
49 |
+
else:
|
50 |
+
from .pipeline_spectrogram_diffusion import SpectrogramDiffusionPipeline
|
51 |
+
from .pipeline_spectrogram_diffusion import SpectrogramContEncoder
|
52 |
+
from .pipeline_spectrogram_diffusion import SpectrogramNotesEncoder
|
53 |
+
from .pipeline_spectrogram_diffusion import T5FilmDecoder
|
54 |
+
|
55 |
+
try:
|
56 |
+
if not (is_transformers_available() and is_torch_available() and is_note_seq_available()):
|
57 |
+
raise OptionalDependencyNotAvailable()
|
58 |
+
except OptionalDependencyNotAvailable:
|
59 |
+
from ....utils.dummy_transformers_and_torch_and_note_seq_objects import *
|
60 |
+
|
61 |
+
else:
|
62 |
+
from .midi_utils import MidiProcessor
|
63 |
+
|
64 |
+
else:
|
65 |
+
import sys
|
66 |
+
|
67 |
+
sys.modules[__name__] = _LazyModule(
|
68 |
+
__name__,
|
69 |
+
globals()["__file__"],
|
70 |
+
_import_structure,
|
71 |
+
module_spec=__spec__,
|
72 |
+
)
|
73 |
+
|
74 |
+
for name, value in _dummy_objects.items():
|
75 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/stable_diffusion_variants/__init__.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ....utils import dummy_torch_and_transformers_objects
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_cycle_diffusion"] = ["CycleDiffusionPipeline"]
|
25 |
+
_import_structure["pipeline_stable_diffusion_inpaint_legacy"] = ["StableDiffusionInpaintPipelineLegacy"]
|
26 |
+
_import_structure["pipeline_stable_diffusion_model_editing"] = ["StableDiffusionModelEditingPipeline"]
|
27 |
+
|
28 |
+
_import_structure["pipeline_stable_diffusion_paradigms"] = ["StableDiffusionParadigmsPipeline"]
|
29 |
+
_import_structure["pipeline_stable_diffusion_pix2pix_zero"] = ["StableDiffusionPix2PixZeroPipeline"]
|
30 |
+
|
31 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
32 |
+
try:
|
33 |
+
if not (is_transformers_available() and is_torch_available()):
|
34 |
+
raise OptionalDependencyNotAvailable()
|
35 |
+
except OptionalDependencyNotAvailable:
|
36 |
+
from ....utils.dummy_torch_and_transformers_objects import *
|
37 |
+
|
38 |
+
else:
|
39 |
+
from .pipeline_cycle_diffusion import CycleDiffusionPipeline
|
40 |
+
from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy
|
41 |
+
from .pipeline_stable_diffusion_model_editing import StableDiffusionModelEditingPipeline
|
42 |
+
from .pipeline_stable_diffusion_paradigms import StableDiffusionParadigmsPipeline
|
43 |
+
from .pipeline_stable_diffusion_pix2pix_zero import StableDiffusionPix2PixZeroPipeline
|
44 |
+
|
45 |
+
else:
|
46 |
+
import sys
|
47 |
+
|
48 |
+
sys.modules[__name__] = _LazyModule(
|
49 |
+
__name__,
|
50 |
+
globals()["__file__"],
|
51 |
+
_import_structure,
|
52 |
+
module_spec=__spec__,
|
53 |
+
)
|
54 |
+
for name, value in _dummy_objects.items():
|
55 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/stochastic_karras_ve/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_stochastic_karras_ve": ["KarrasVePipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_stochastic_karras_ve import KarrasVePipeline
|
10 |
+
|
11 |
+
else:
|
12 |
+
import sys
|
13 |
+
|
14 |
+
sys.modules[__name__] = _LazyModule(
|
15 |
+
__name__,
|
16 |
+
globals()["__file__"],
|
17 |
+
_import_structure,
|
18 |
+
module_spec=__spec__,
|
19 |
+
)
|
diffusers/pipelines/deprecated/versatile_diffusion/__init__.py
ADDED
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
is_torch_available,
|
8 |
+
is_transformers_available,
|
9 |
+
is_transformers_version,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ....utils.dummy_torch_and_transformers_objects import (
|
21 |
+
VersatileDiffusionDualGuidedPipeline,
|
22 |
+
VersatileDiffusionImageVariationPipeline,
|
23 |
+
VersatileDiffusionPipeline,
|
24 |
+
VersatileDiffusionTextToImagePipeline,
|
25 |
+
)
|
26 |
+
|
27 |
+
_dummy_objects.update(
|
28 |
+
{
|
29 |
+
"VersatileDiffusionDualGuidedPipeline": VersatileDiffusionDualGuidedPipeline,
|
30 |
+
"VersatileDiffusionImageVariationPipeline": VersatileDiffusionImageVariationPipeline,
|
31 |
+
"VersatileDiffusionPipeline": VersatileDiffusionPipeline,
|
32 |
+
"VersatileDiffusionTextToImagePipeline": VersatileDiffusionTextToImagePipeline,
|
33 |
+
}
|
34 |
+
)
|
35 |
+
else:
|
36 |
+
_import_structure["modeling_text_unet"] = ["UNetFlatConditionModel"]
|
37 |
+
_import_structure["pipeline_versatile_diffusion"] = ["VersatileDiffusionPipeline"]
|
38 |
+
_import_structure["pipeline_versatile_diffusion_dual_guided"] = ["VersatileDiffusionDualGuidedPipeline"]
|
39 |
+
_import_structure["pipeline_versatile_diffusion_image_variation"] = ["VersatileDiffusionImageVariationPipeline"]
|
40 |
+
_import_structure["pipeline_versatile_diffusion_text_to_image"] = ["VersatileDiffusionTextToImagePipeline"]
|
41 |
+
|
42 |
+
|
43 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
44 |
+
try:
|
45 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
|
46 |
+
raise OptionalDependencyNotAvailable()
|
47 |
+
except OptionalDependencyNotAvailable:
|
48 |
+
from ....utils.dummy_torch_and_transformers_objects import (
|
49 |
+
VersatileDiffusionDualGuidedPipeline,
|
50 |
+
VersatileDiffusionImageVariationPipeline,
|
51 |
+
VersatileDiffusionPipeline,
|
52 |
+
VersatileDiffusionTextToImagePipeline,
|
53 |
+
)
|
54 |
+
else:
|
55 |
+
from .pipeline_versatile_diffusion import VersatileDiffusionPipeline
|
56 |
+
from .pipeline_versatile_diffusion_dual_guided import VersatileDiffusionDualGuidedPipeline
|
57 |
+
from .pipeline_versatile_diffusion_image_variation import VersatileDiffusionImageVariationPipeline
|
58 |
+
from .pipeline_versatile_diffusion_text_to_image import VersatileDiffusionTextToImagePipeline
|
59 |
+
|
60 |
+
else:
|
61 |
+
import sys
|
62 |
+
|
63 |
+
sys.modules[__name__] = _LazyModule(
|
64 |
+
__name__,
|
65 |
+
globals()["__file__"],
|
66 |
+
_import_structure,
|
67 |
+
module_spec=__spec__,
|
68 |
+
)
|
69 |
+
|
70 |
+
for name, value in _dummy_objects.items():
|
71 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/deprecated/vq_diffusion/__init__.py
ADDED
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ....utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
is_torch_available,
|
8 |
+
is_transformers_available,
|
9 |
+
)
|
10 |
+
|
11 |
+
|
12 |
+
_dummy_objects = {}
|
13 |
+
_import_structure = {}
|
14 |
+
|
15 |
+
try:
|
16 |
+
if not (is_transformers_available() and is_torch_available()):
|
17 |
+
raise OptionalDependencyNotAvailable()
|
18 |
+
except OptionalDependencyNotAvailable:
|
19 |
+
from ....utils.dummy_torch_and_transformers_objects import (
|
20 |
+
LearnedClassifierFreeSamplingEmbeddings,
|
21 |
+
VQDiffusionPipeline,
|
22 |
+
)
|
23 |
+
|
24 |
+
_dummy_objects.update(
|
25 |
+
{
|
26 |
+
"LearnedClassifierFreeSamplingEmbeddings": LearnedClassifierFreeSamplingEmbeddings,
|
27 |
+
"VQDiffusionPipeline": VQDiffusionPipeline,
|
28 |
+
}
|
29 |
+
)
|
30 |
+
else:
|
31 |
+
_import_structure["pipeline_vq_diffusion"] = ["LearnedClassifierFreeSamplingEmbeddings", "VQDiffusionPipeline"]
|
32 |
+
|
33 |
+
|
34 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
35 |
+
try:
|
36 |
+
if not (is_transformers_available() and is_torch_available()):
|
37 |
+
raise OptionalDependencyNotAvailable()
|
38 |
+
except OptionalDependencyNotAvailable:
|
39 |
+
from ....utils.dummy_torch_and_transformers_objects import (
|
40 |
+
LearnedClassifierFreeSamplingEmbeddings,
|
41 |
+
VQDiffusionPipeline,
|
42 |
+
)
|
43 |
+
else:
|
44 |
+
from .pipeline_vq_diffusion import LearnedClassifierFreeSamplingEmbeddings, VQDiffusionPipeline
|
45 |
+
|
46 |
+
else:
|
47 |
+
import sys
|
48 |
+
|
49 |
+
sys.modules[__name__] = _LazyModule(
|
50 |
+
__name__,
|
51 |
+
globals()["__file__"],
|
52 |
+
_import_structure,
|
53 |
+
module_spec=__spec__,
|
54 |
+
)
|
55 |
+
|
56 |
+
for name, value in _dummy_objects.items():
|
57 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/dit/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import DIFFUSERS_SLOW_IMPORT, _LazyModule
|
4 |
+
|
5 |
+
|
6 |
+
_import_structure = {"pipeline_dit": ["DiTPipeline"]}
|
7 |
+
|
8 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
9 |
+
from .pipeline_dit import DiTPipeline
|
10 |
+
|
11 |
+
else:
|
12 |
+
import sys
|
13 |
+
|
14 |
+
sys.modules[__name__] = _LazyModule(
|
15 |
+
__name__,
|
16 |
+
globals()["__file__"],
|
17 |
+
_import_structure,
|
18 |
+
module_spec=__spec__,
|
19 |
+
)
|
diffusers/pipelines/i2vgen_xl/__init__.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_i2vgen_xl"] = ["I2VGenXLPipeline"]
|
25 |
+
|
26 |
+
|
27 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
28 |
+
try:
|
29 |
+
if not (is_transformers_available() and is_torch_available()):
|
30 |
+
raise OptionalDependencyNotAvailable()
|
31 |
+
except OptionalDependencyNotAvailable:
|
32 |
+
from ...utils.dummy_torch_and_transformers_objects import * # noqa F403
|
33 |
+
else:
|
34 |
+
from .pipeline_i2vgen_xl import I2VGenXLPipeline
|
35 |
+
|
36 |
+
else:
|
37 |
+
import sys
|
38 |
+
|
39 |
+
sys.modules[__name__] = _LazyModule(
|
40 |
+
__name__,
|
41 |
+
globals()["__file__"],
|
42 |
+
_import_structure,
|
43 |
+
module_spec=__spec__,
|
44 |
+
)
|
45 |
+
for name, value in _dummy_objects.items():
|
46 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/kandinsky/__init__.py
ADDED
@@ -0,0 +1,66 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_kandinsky"] = ["KandinskyPipeline"]
|
25 |
+
_import_structure["pipeline_kandinsky_combined"] = [
|
26 |
+
"KandinskyCombinedPipeline",
|
27 |
+
"KandinskyImg2ImgCombinedPipeline",
|
28 |
+
"KandinskyInpaintCombinedPipeline",
|
29 |
+
]
|
30 |
+
_import_structure["pipeline_kandinsky_img2img"] = ["KandinskyImg2ImgPipeline"]
|
31 |
+
_import_structure["pipeline_kandinsky_inpaint"] = ["KandinskyInpaintPipeline"]
|
32 |
+
_import_structure["pipeline_kandinsky_prior"] = ["KandinskyPriorPipeline", "KandinskyPriorPipelineOutput"]
|
33 |
+
_import_structure["text_encoder"] = ["MultilingualCLIP"]
|
34 |
+
|
35 |
+
|
36 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
37 |
+
try:
|
38 |
+
if not (is_transformers_available() and is_torch_available()):
|
39 |
+
raise OptionalDependencyNotAvailable()
|
40 |
+
except OptionalDependencyNotAvailable:
|
41 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
42 |
+
|
43 |
+
else:
|
44 |
+
from .pipeline_kandinsky import KandinskyPipeline
|
45 |
+
from .pipeline_kandinsky_combined import (
|
46 |
+
KandinskyCombinedPipeline,
|
47 |
+
KandinskyImg2ImgCombinedPipeline,
|
48 |
+
KandinskyInpaintCombinedPipeline,
|
49 |
+
)
|
50 |
+
from .pipeline_kandinsky_img2img import KandinskyImg2ImgPipeline
|
51 |
+
from .pipeline_kandinsky_inpaint import KandinskyInpaintPipeline
|
52 |
+
from .pipeline_kandinsky_prior import KandinskyPriorPipeline, KandinskyPriorPipelineOutput
|
53 |
+
from .text_encoder import MultilingualCLIP
|
54 |
+
|
55 |
+
else:
|
56 |
+
import sys
|
57 |
+
|
58 |
+
sys.modules[__name__] = _LazyModule(
|
59 |
+
__name__,
|
60 |
+
globals()["__file__"],
|
61 |
+
_import_structure,
|
62 |
+
module_spec=__spec__,
|
63 |
+
)
|
64 |
+
|
65 |
+
for name, value in _dummy_objects.items():
|
66 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/kandinsky2_2/__init__.py
ADDED
@@ -0,0 +1,70 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_kandinsky2_2"] = ["KandinskyV22Pipeline"]
|
25 |
+
_import_structure["pipeline_kandinsky2_2_combined"] = [
|
26 |
+
"KandinskyV22CombinedPipeline",
|
27 |
+
"KandinskyV22Img2ImgCombinedPipeline",
|
28 |
+
"KandinskyV22InpaintCombinedPipeline",
|
29 |
+
]
|
30 |
+
_import_structure["pipeline_kandinsky2_2_controlnet"] = ["KandinskyV22ControlnetPipeline"]
|
31 |
+
_import_structure["pipeline_kandinsky2_2_controlnet_img2img"] = ["KandinskyV22ControlnetImg2ImgPipeline"]
|
32 |
+
_import_structure["pipeline_kandinsky2_2_img2img"] = ["KandinskyV22Img2ImgPipeline"]
|
33 |
+
_import_structure["pipeline_kandinsky2_2_inpainting"] = ["KandinskyV22InpaintPipeline"]
|
34 |
+
_import_structure["pipeline_kandinsky2_2_prior"] = ["KandinskyV22PriorPipeline"]
|
35 |
+
_import_structure["pipeline_kandinsky2_2_prior_emb2emb"] = ["KandinskyV22PriorEmb2EmbPipeline"]
|
36 |
+
|
37 |
+
|
38 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
39 |
+
try:
|
40 |
+
if not (is_transformers_available() and is_torch_available()):
|
41 |
+
raise OptionalDependencyNotAvailable()
|
42 |
+
|
43 |
+
except OptionalDependencyNotAvailable:
|
44 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
45 |
+
else:
|
46 |
+
from .pipeline_kandinsky2_2 import KandinskyV22Pipeline
|
47 |
+
from .pipeline_kandinsky2_2_combined import (
|
48 |
+
KandinskyV22CombinedPipeline,
|
49 |
+
KandinskyV22Img2ImgCombinedPipeline,
|
50 |
+
KandinskyV22InpaintCombinedPipeline,
|
51 |
+
)
|
52 |
+
from .pipeline_kandinsky2_2_controlnet import KandinskyV22ControlnetPipeline
|
53 |
+
from .pipeline_kandinsky2_2_controlnet_img2img import KandinskyV22ControlnetImg2ImgPipeline
|
54 |
+
from .pipeline_kandinsky2_2_img2img import KandinskyV22Img2ImgPipeline
|
55 |
+
from .pipeline_kandinsky2_2_inpainting import KandinskyV22InpaintPipeline
|
56 |
+
from .pipeline_kandinsky2_2_prior import KandinskyV22PriorPipeline
|
57 |
+
from .pipeline_kandinsky2_2_prior_emb2emb import KandinskyV22PriorEmb2EmbPipeline
|
58 |
+
|
59 |
+
else:
|
60 |
+
import sys
|
61 |
+
|
62 |
+
sys.modules[__name__] = _LazyModule(
|
63 |
+
__name__,
|
64 |
+
globals()["__file__"],
|
65 |
+
_import_structure,
|
66 |
+
module_spec=__spec__,
|
67 |
+
)
|
68 |
+
|
69 |
+
for name, value in _dummy_objects.items():
|
70 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/kandinsky3/__init__.py
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_kandinsky3"] = ["Kandinsky3Pipeline"]
|
25 |
+
_import_structure["pipeline_kandinsky3_img2img"] = ["Kandinsky3Img2ImgPipeline"]
|
26 |
+
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available()):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_kandinsky3 import Kandinsky3Pipeline
|
37 |
+
from .pipeline_kandinsky3_img2img import Kandinsky3Img2ImgPipeline
|
38 |
+
else:
|
39 |
+
import sys
|
40 |
+
|
41 |
+
sys.modules[__name__] = _LazyModule(
|
42 |
+
__name__,
|
43 |
+
globals()["__file__"],
|
44 |
+
_import_structure,
|
45 |
+
module_spec=__spec__,
|
46 |
+
)
|
47 |
+
|
48 |
+
for name, value in _dummy_objects.items():
|
49 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/latent_consistency_models/__init__.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available()):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["pipeline_latent_consistency_img2img"] = ["LatentConsistencyModelImg2ImgPipeline"]
|
26 |
+
_import_structure["pipeline_latent_consistency_text2img"] = ["LatentConsistencyModelPipeline"]
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available()):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_latent_consistency_img2img import LatentConsistencyModelImg2ImgPipeline
|
37 |
+
from .pipeline_latent_consistency_text2img import LatentConsistencyModelPipeline
|
38 |
+
|
39 |
+
else:
|
40 |
+
import sys
|
41 |
+
|
42 |
+
sys.modules[__name__] = _LazyModule(
|
43 |
+
__name__,
|
44 |
+
globals()["__file__"],
|
45 |
+
_import_structure,
|
46 |
+
module_spec=__spec__,
|
47 |
+
)
|
48 |
+
|
49 |
+
for name, value in _dummy_objects.items():
|
50 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/latent_diffusion/__init__.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_latent_diffusion"] = ["LDMBertModel", "LDMTextToImagePipeline"]
|
25 |
+
_import_structure["pipeline_latent_diffusion_superresolution"] = ["LDMSuperResolutionPipeline"]
|
26 |
+
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available()):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_latent_diffusion import LDMBertModel, LDMTextToImagePipeline
|
37 |
+
from .pipeline_latent_diffusion_superresolution import LDMSuperResolutionPipeline
|
38 |
+
|
39 |
+
else:
|
40 |
+
import sys
|
41 |
+
|
42 |
+
sys.modules[__name__] = _LazyModule(
|
43 |
+
__name__,
|
44 |
+
globals()["__file__"],
|
45 |
+
_import_structure,
|
46 |
+
module_spec=__spec__,
|
47 |
+
)
|
48 |
+
|
49 |
+
for name, value in _dummy_objects.items():
|
50 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/ledits_pp/__init__.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_leditspp_stable_diffusion"] = ["LEditsPPPipelineStableDiffusion"]
|
25 |
+
_import_structure["pipeline_leditspp_stable_diffusion_xl"] = ["LEditsPPPipelineStableDiffusionXL"]
|
26 |
+
|
27 |
+
_import_structure["pipeline_output"] = ["LEditsPPDiffusionPipelineOutput", "LEditsPPDiffusionPipelineOutput"]
|
28 |
+
|
29 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
30 |
+
try:
|
31 |
+
if not (is_transformers_available() and is_torch_available()):
|
32 |
+
raise OptionalDependencyNotAvailable()
|
33 |
+
|
34 |
+
except OptionalDependencyNotAvailable:
|
35 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
36 |
+
else:
|
37 |
+
from .pipeline_leditspp_stable_diffusion import (
|
38 |
+
LEditsPPDiffusionPipelineOutput,
|
39 |
+
LEditsPPInversionPipelineOutput,
|
40 |
+
LEditsPPPipelineStableDiffusion,
|
41 |
+
)
|
42 |
+
from .pipeline_leditspp_stable_diffusion_xl import LEditsPPPipelineStableDiffusionXL
|
43 |
+
|
44 |
+
else:
|
45 |
+
import sys
|
46 |
+
|
47 |
+
sys.modules[__name__] = _LazyModule(
|
48 |
+
__name__,
|
49 |
+
globals()["__file__"],
|
50 |
+
_import_structure,
|
51 |
+
module_spec=__spec__,
|
52 |
+
)
|
53 |
+
|
54 |
+
for name, value in _dummy_objects.items():
|
55 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/musicldm/__init__.py
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
is_transformers_version,
|
11 |
+
)
|
12 |
+
|
13 |
+
|
14 |
+
_dummy_objects = {}
|
15 |
+
_import_structure = {}
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["pipeline_musicldm"] = ["MusicLDMPipeline"]
|
26 |
+
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.27.0")):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_musicldm import MusicLDMPipeline
|
37 |
+
|
38 |
+
else:
|
39 |
+
import sys
|
40 |
+
|
41 |
+
sys.modules[__name__] = _LazyModule(
|
42 |
+
__name__,
|
43 |
+
globals()["__file__"],
|
44 |
+
_import_structure,
|
45 |
+
module_spec=__spec__,
|
46 |
+
)
|
47 |
+
|
48 |
+
for name, value in _dummy_objects.items():
|
49 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/paint_by_example/__init__.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from dataclasses import dataclass
|
2 |
+
from typing import TYPE_CHECKING, List, Optional, Union
|
3 |
+
|
4 |
+
import numpy as np
|
5 |
+
import PIL
|
6 |
+
from PIL import Image
|
7 |
+
|
8 |
+
from ...utils import (
|
9 |
+
DIFFUSERS_SLOW_IMPORT,
|
10 |
+
OptionalDependencyNotAvailable,
|
11 |
+
_LazyModule,
|
12 |
+
get_objects_from_module,
|
13 |
+
is_torch_available,
|
14 |
+
is_transformers_available,
|
15 |
+
)
|
16 |
+
|
17 |
+
|
18 |
+
_dummy_objects = {}
|
19 |
+
_import_structure = {}
|
20 |
+
|
21 |
+
try:
|
22 |
+
if not (is_transformers_available() and is_torch_available()):
|
23 |
+
raise OptionalDependencyNotAvailable()
|
24 |
+
except OptionalDependencyNotAvailable:
|
25 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
26 |
+
|
27 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
28 |
+
else:
|
29 |
+
_import_structure["image_encoder"] = ["PaintByExampleImageEncoder"]
|
30 |
+
_import_structure["pipeline_paint_by_example"] = ["PaintByExamplePipeline"]
|
31 |
+
|
32 |
+
|
33 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
34 |
+
try:
|
35 |
+
if not (is_transformers_available() and is_torch_available()):
|
36 |
+
raise OptionalDependencyNotAvailable()
|
37 |
+
|
38 |
+
except OptionalDependencyNotAvailable:
|
39 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
40 |
+
else:
|
41 |
+
from .image_encoder import PaintByExampleImageEncoder
|
42 |
+
from .pipeline_paint_by_example import PaintByExamplePipeline
|
43 |
+
|
44 |
+
else:
|
45 |
+
import sys
|
46 |
+
|
47 |
+
sys.modules[__name__] = _LazyModule(
|
48 |
+
__name__,
|
49 |
+
globals()["__file__"],
|
50 |
+
_import_structure,
|
51 |
+
module_spec=__spec__,
|
52 |
+
)
|
53 |
+
|
54 |
+
for name, value in _dummy_objects.items():
|
55 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/pia/__init__.py
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_pia"] = ["PIAPipeline", "PIAPipelineOutput"]
|
25 |
+
|
26 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
27 |
+
try:
|
28 |
+
if not (is_transformers_available() and is_torch_available()):
|
29 |
+
raise OptionalDependencyNotAvailable()
|
30 |
+
except OptionalDependencyNotAvailable:
|
31 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
32 |
+
|
33 |
+
else:
|
34 |
+
from .pipeline_pia import PIAPipeline, PIAPipelineOutput
|
35 |
+
|
36 |
+
else:
|
37 |
+
import sys
|
38 |
+
|
39 |
+
sys.modules[__name__] = _LazyModule(
|
40 |
+
__name__,
|
41 |
+
globals()["__file__"],
|
42 |
+
_import_structure,
|
43 |
+
module_spec=__spec__,
|
44 |
+
)
|
45 |
+
for name, value in _dummy_objects.items():
|
46 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/pixart_alpha/__init__.py
ADDED
@@ -0,0 +1,55 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
|
17 |
+
try:
|
18 |
+
if not (is_transformers_available() and is_torch_available()):
|
19 |
+
raise OptionalDependencyNotAvailable()
|
20 |
+
except OptionalDependencyNotAvailable:
|
21 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
22 |
+
|
23 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
24 |
+
else:
|
25 |
+
_import_structure["pipeline_pixart_alpha"] = ["PixArtAlphaPipeline"]
|
26 |
+
_import_structure["pipeline_pixart_sigma"] = ["PixArtSigmaPipeline"]
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available()):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_pixart_alpha import (
|
37 |
+
ASPECT_RATIO_256_BIN,
|
38 |
+
ASPECT_RATIO_512_BIN,
|
39 |
+
ASPECT_RATIO_1024_BIN,
|
40 |
+
PixArtAlphaPipeline,
|
41 |
+
)
|
42 |
+
from .pipeline_pixart_sigma import ASPECT_RATIO_2048_BIN, PixArtSigmaPipeline
|
43 |
+
|
44 |
+
else:
|
45 |
+
import sys
|
46 |
+
|
47 |
+
sys.modules[__name__] = _LazyModule(
|
48 |
+
__name__,
|
49 |
+
globals()["__file__"],
|
50 |
+
_import_structure,
|
51 |
+
module_spec=__spec__,
|
52 |
+
)
|
53 |
+
|
54 |
+
for name, value in _dummy_objects.items():
|
55 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/semantic_stable_diffusion/__init__.py
ADDED
@@ -0,0 +1,49 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_output"] = ["SemanticStableDiffusionPipelineOutput"]
|
25 |
+
_import_structure["pipeline_semantic_stable_diffusion"] = ["SemanticStableDiffusionPipeline"]
|
26 |
+
|
27 |
+
|
28 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
29 |
+
try:
|
30 |
+
if not (is_transformers_available() and is_torch_available()):
|
31 |
+
raise OptionalDependencyNotAvailable()
|
32 |
+
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
35 |
+
else:
|
36 |
+
from .pipeline_semantic_stable_diffusion import SemanticStableDiffusionPipeline
|
37 |
+
|
38 |
+
else:
|
39 |
+
import sys
|
40 |
+
|
41 |
+
sys.modules[__name__] = _LazyModule(
|
42 |
+
__name__,
|
43 |
+
globals()["__file__"],
|
44 |
+
_import_structure,
|
45 |
+
module_spec=__spec__,
|
46 |
+
)
|
47 |
+
|
48 |
+
for name, value in _dummy_objects.items():
|
49 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/shap_e/__init__.py
ADDED
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["camera"] = ["create_pan_cameras"]
|
25 |
+
_import_structure["pipeline_shap_e"] = ["ShapEPipeline"]
|
26 |
+
_import_structure["pipeline_shap_e_img2img"] = ["ShapEImg2ImgPipeline"]
|
27 |
+
_import_structure["renderer"] = [
|
28 |
+
"BoundingBoxVolume",
|
29 |
+
"ImportanceRaySampler",
|
30 |
+
"MLPNeRFModelOutput",
|
31 |
+
"MLPNeRSTFModel",
|
32 |
+
"ShapEParamsProjModel",
|
33 |
+
"ShapERenderer",
|
34 |
+
"StratifiedRaySampler",
|
35 |
+
"VoidNeRFModel",
|
36 |
+
]
|
37 |
+
|
38 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
39 |
+
try:
|
40 |
+
if not (is_transformers_available() and is_torch_available()):
|
41 |
+
raise OptionalDependencyNotAvailable()
|
42 |
+
|
43 |
+
except OptionalDependencyNotAvailable:
|
44 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
45 |
+
else:
|
46 |
+
from .camera import create_pan_cameras
|
47 |
+
from .pipeline_shap_e import ShapEPipeline
|
48 |
+
from .pipeline_shap_e_img2img import ShapEImg2ImgPipeline
|
49 |
+
from .renderer import (
|
50 |
+
BoundingBoxVolume,
|
51 |
+
ImportanceRaySampler,
|
52 |
+
MLPNeRFModelOutput,
|
53 |
+
MLPNeRSTFModel,
|
54 |
+
ShapEParamsProjModel,
|
55 |
+
ShapERenderer,
|
56 |
+
StratifiedRaySampler,
|
57 |
+
VoidNeRFModel,
|
58 |
+
)
|
59 |
+
|
60 |
+
else:
|
61 |
+
import sys
|
62 |
+
|
63 |
+
sys.modules[__name__] = _LazyModule(
|
64 |
+
__name__,
|
65 |
+
globals()["__file__"],
|
66 |
+
_import_structure,
|
67 |
+
module_spec=__spec__,
|
68 |
+
)
|
69 |
+
|
70 |
+
for name, value in _dummy_objects.items():
|
71 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/stable_cascade/__init__.py
ADDED
@@ -0,0 +1,50 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_torch_available,
|
9 |
+
is_transformers_available,
|
10 |
+
)
|
11 |
+
|
12 |
+
|
13 |
+
_dummy_objects = {}
|
14 |
+
_import_structure = {}
|
15 |
+
|
16 |
+
try:
|
17 |
+
if not (is_transformers_available() and is_torch_available()):
|
18 |
+
raise OptionalDependencyNotAvailable()
|
19 |
+
except OptionalDependencyNotAvailable:
|
20 |
+
from ...utils import dummy_torch_and_transformers_objects
|
21 |
+
|
22 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
23 |
+
else:
|
24 |
+
_import_structure["pipeline_stable_cascade"] = ["StableCascadeDecoderPipeline"]
|
25 |
+
_import_structure["pipeline_stable_cascade_combined"] = ["StableCascadeCombinedPipeline"]
|
26 |
+
_import_structure["pipeline_stable_cascade_prior"] = ["StableCascadePriorPipeline"]
|
27 |
+
|
28 |
+
|
29 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
30 |
+
try:
|
31 |
+
if not (is_transformers_available() and is_torch_available()):
|
32 |
+
raise OptionalDependencyNotAvailable()
|
33 |
+
except OptionalDependencyNotAvailable:
|
34 |
+
from ...utils.dummy_torch_and_transformers_objects import * # noqa F403
|
35 |
+
else:
|
36 |
+
from .pipeline_stable_cascade import StableCascadeDecoderPipeline
|
37 |
+
from .pipeline_stable_cascade_combined import StableCascadeCombinedPipeline
|
38 |
+
from .pipeline_stable_cascade_prior import StableCascadePriorPipeline
|
39 |
+
else:
|
40 |
+
import sys
|
41 |
+
|
42 |
+
sys.modules[__name__] = _LazyModule(
|
43 |
+
__name__,
|
44 |
+
globals()["__file__"],
|
45 |
+
_import_structure,
|
46 |
+
module_spec=__spec__,
|
47 |
+
)
|
48 |
+
|
49 |
+
for name, value in _dummy_objects.items():
|
50 |
+
setattr(sys.modules[__name__], name, value)
|
diffusers/pipelines/stable_diffusion/__init__.py
ADDED
@@ -0,0 +1,203 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import TYPE_CHECKING
|
2 |
+
|
3 |
+
from ...utils import (
|
4 |
+
DIFFUSERS_SLOW_IMPORT,
|
5 |
+
OptionalDependencyNotAvailable,
|
6 |
+
_LazyModule,
|
7 |
+
get_objects_from_module,
|
8 |
+
is_flax_available,
|
9 |
+
is_k_diffusion_available,
|
10 |
+
is_k_diffusion_version,
|
11 |
+
is_onnx_available,
|
12 |
+
is_torch_available,
|
13 |
+
is_transformers_available,
|
14 |
+
is_transformers_version,
|
15 |
+
)
|
16 |
+
|
17 |
+
|
18 |
+
_dummy_objects = {}
|
19 |
+
_additional_imports = {}
|
20 |
+
_import_structure = {"pipeline_output": ["StableDiffusionPipelineOutput"]}
|
21 |
+
|
22 |
+
if is_transformers_available() and is_flax_available():
|
23 |
+
_import_structure["pipeline_output"].extend(["FlaxStableDiffusionPipelineOutput"])
|
24 |
+
try:
|
25 |
+
if not (is_transformers_available() and is_torch_available()):
|
26 |
+
raise OptionalDependencyNotAvailable()
|
27 |
+
except OptionalDependencyNotAvailable:
|
28 |
+
from ...utils import dummy_torch_and_transformers_objects # noqa F403
|
29 |
+
|
30 |
+
_dummy_objects.update(get_objects_from_module(dummy_torch_and_transformers_objects))
|
31 |
+
else:
|
32 |
+
_import_structure["clip_image_project_model"] = ["CLIPImageProjection"]
|
33 |
+
_import_structure["pipeline_cycle_diffusion"] = ["CycleDiffusionPipeline"]
|
34 |
+
_import_structure["pipeline_stable_diffusion"] = ["StableDiffusionPipeline"]
|
35 |
+
_import_structure["pipeline_stable_diffusion_attend_and_excite"] = ["StableDiffusionAttendAndExcitePipeline"]
|
36 |
+
_import_structure["pipeline_stable_diffusion_gligen"] = ["StableDiffusionGLIGENPipeline"]
|
37 |
+
_import_structure["pipeline_stable_diffusion_gligen_text_image"] = ["StableDiffusionGLIGENTextImagePipeline"]
|
38 |
+
_import_structure["pipeline_stable_diffusion_img2img"] = ["StableDiffusionImg2ImgPipeline"]
|
39 |
+
_import_structure["pipeline_stable_diffusion_inpaint"] = ["StableDiffusionInpaintPipeline"]
|
40 |
+
_import_structure["pipeline_stable_diffusion_inpaint_legacy"] = ["StableDiffusionInpaintPipelineLegacy"]
|
41 |
+
_import_structure["pipeline_stable_diffusion_instruct_pix2pix"] = ["StableDiffusionInstructPix2PixPipeline"]
|
42 |
+
_import_structure["pipeline_stable_diffusion_latent_upscale"] = ["StableDiffusionLatentUpscalePipeline"]
|
43 |
+
_import_structure["pipeline_stable_diffusion_model_editing"] = ["StableDiffusionModelEditingPipeline"]
|
44 |
+
_import_structure["pipeline_stable_diffusion_paradigms"] = ["StableDiffusionParadigmsPipeline"]
|
45 |
+
_import_structure["pipeline_stable_diffusion_upscale"] = ["StableDiffusionUpscalePipeline"]
|
46 |
+
_import_structure["pipeline_stable_unclip"] = ["StableUnCLIPPipeline"]
|
47 |
+
_import_structure["pipeline_stable_unclip_img2img"] = ["StableUnCLIPImg2ImgPipeline"]
|
48 |
+
_import_structure["safety_checker"] = ["StableDiffusionSafetyChecker"]
|
49 |
+
_import_structure["stable_unclip_image_normalizer"] = ["StableUnCLIPImageNormalizer"]
|
50 |
+
try:
|
51 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
|
52 |
+
raise OptionalDependencyNotAvailable()
|
53 |
+
except OptionalDependencyNotAvailable:
|
54 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
55 |
+
StableDiffusionImageVariationPipeline,
|
56 |
+
)
|
57 |
+
|
58 |
+
_dummy_objects.update({"StableDiffusionImageVariationPipeline": StableDiffusionImageVariationPipeline})
|
59 |
+
else:
|
60 |
+
_import_structure["pipeline_stable_diffusion_image_variation"] = ["StableDiffusionImageVariationPipeline"]
|
61 |
+
try:
|
62 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.26.0")):
|
63 |
+
raise OptionalDependencyNotAvailable()
|
64 |
+
except OptionalDependencyNotAvailable:
|
65 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
66 |
+
StableDiffusionDepth2ImgPipeline,
|
67 |
+
)
|
68 |
+
|
69 |
+
_dummy_objects.update(
|
70 |
+
{
|
71 |
+
"StableDiffusionDepth2ImgPipeline": StableDiffusionDepth2ImgPipeline,
|
72 |
+
}
|
73 |
+
)
|
74 |
+
else:
|
75 |
+
_import_structure["pipeline_stable_diffusion_depth2img"] = ["StableDiffusionDepth2ImgPipeline"]
|
76 |
+
|
77 |
+
try:
|
78 |
+
if not (is_transformers_available() and is_onnx_available()):
|
79 |
+
raise OptionalDependencyNotAvailable()
|
80 |
+
except OptionalDependencyNotAvailable:
|
81 |
+
from ...utils import dummy_onnx_objects # noqa F403
|
82 |
+
|
83 |
+
_dummy_objects.update(get_objects_from_module(dummy_onnx_objects))
|
84 |
+
else:
|
85 |
+
_import_structure["pipeline_onnx_stable_diffusion"] = [
|
86 |
+
"OnnxStableDiffusionPipeline",
|
87 |
+
"StableDiffusionOnnxPipeline",
|
88 |
+
]
|
89 |
+
_import_structure["pipeline_onnx_stable_diffusion_img2img"] = ["OnnxStableDiffusionImg2ImgPipeline"]
|
90 |
+
_import_structure["pipeline_onnx_stable_diffusion_inpaint"] = ["OnnxStableDiffusionInpaintPipeline"]
|
91 |
+
_import_structure["pipeline_onnx_stable_diffusion_inpaint_legacy"] = ["OnnxStableDiffusionInpaintPipelineLegacy"]
|
92 |
+
_import_structure["pipeline_onnx_stable_diffusion_upscale"] = ["OnnxStableDiffusionUpscalePipeline"]
|
93 |
+
|
94 |
+
if is_transformers_available() and is_flax_available():
|
95 |
+
from ...schedulers.scheduling_pndm_flax import PNDMSchedulerState
|
96 |
+
|
97 |
+
_additional_imports.update({"PNDMSchedulerState": PNDMSchedulerState})
|
98 |
+
_import_structure["pipeline_flax_stable_diffusion"] = ["FlaxStableDiffusionPipeline"]
|
99 |
+
_import_structure["pipeline_flax_stable_diffusion_img2img"] = ["FlaxStableDiffusionImg2ImgPipeline"]
|
100 |
+
_import_structure["pipeline_flax_stable_diffusion_inpaint"] = ["FlaxStableDiffusionInpaintPipeline"]
|
101 |
+
_import_structure["safety_checker_flax"] = ["FlaxStableDiffusionSafetyChecker"]
|
102 |
+
|
103 |
+
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
|
104 |
+
try:
|
105 |
+
if not (is_transformers_available() and is_torch_available()):
|
106 |
+
raise OptionalDependencyNotAvailable()
|
107 |
+
|
108 |
+
except OptionalDependencyNotAvailable:
|
109 |
+
from ...utils.dummy_torch_and_transformers_objects import *
|
110 |
+
|
111 |
+
else:
|
112 |
+
from .clip_image_project_model import CLIPImageProjection
|
113 |
+
from .pipeline_stable_diffusion import (
|
114 |
+
StableDiffusionPipeline,
|
115 |
+
StableDiffusionPipelineOutput,
|
116 |
+
StableDiffusionSafetyChecker,
|
117 |
+
)
|
118 |
+
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
|
119 |
+
from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
|
120 |
+
from .pipeline_stable_diffusion_instruct_pix2pix import (
|
121 |
+
StableDiffusionInstructPix2PixPipeline,
|
122 |
+
)
|
123 |
+
from .pipeline_stable_diffusion_latent_upscale import (
|
124 |
+
StableDiffusionLatentUpscalePipeline,
|
125 |
+
)
|
126 |
+
from .pipeline_stable_diffusion_upscale import StableDiffusionUpscalePipeline
|
127 |
+
from .pipeline_stable_unclip import StableUnCLIPPipeline
|
128 |
+
from .pipeline_stable_unclip_img2img import StableUnCLIPImg2ImgPipeline
|
129 |
+
from .safety_checker import StableDiffusionSafetyChecker
|
130 |
+
from .stable_unclip_image_normalizer import StableUnCLIPImageNormalizer
|
131 |
+
|
132 |
+
try:
|
133 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.25.0")):
|
134 |
+
raise OptionalDependencyNotAvailable()
|
135 |
+
except OptionalDependencyNotAvailable:
|
136 |
+
from ...utils.dummy_torch_and_transformers_objects import (
|
137 |
+
StableDiffusionImageVariationPipeline,
|
138 |
+
)
|
139 |
+
else:
|
140 |
+
from .pipeline_stable_diffusion_image_variation import (
|
141 |
+
StableDiffusionImageVariationPipeline,
|
142 |
+
)
|
143 |
+
|
144 |
+
try:
|
145 |
+
if not (is_transformers_available() and is_torch_available() and is_transformers_version(">=", "4.26.0")):
|
146 |
+
raise OptionalDependencyNotAvailable()
|
147 |
+
except OptionalDependencyNotAvailable:
|
148 |
+
from ...utils.dummy_torch_and_transformers_objects import StableDiffusionDepth2ImgPipeline
|
149 |
+
else:
|
150 |
+
from .pipeline_stable_diffusion_depth2img import (
|
151 |
+
StableDiffusionDepth2ImgPipeline,
|
152 |
+
)
|
153 |
+
|
154 |
+
try:
|
155 |
+
if not (is_transformers_available() and is_onnx_available()):
|
156 |
+
raise OptionalDependencyNotAvailable()
|
157 |
+
except OptionalDependencyNotAvailable:
|
158 |
+
from ...utils.dummy_onnx_objects import *
|
159 |
+
else:
|
160 |
+
from .pipeline_onnx_stable_diffusion import (
|
161 |
+
OnnxStableDiffusionPipeline,
|
162 |
+
StableDiffusionOnnxPipeline,
|
163 |
+
)
|
164 |
+
from .pipeline_onnx_stable_diffusion_img2img import (
|
165 |
+
OnnxStableDiffusionImg2ImgPipeline,
|
166 |
+
)
|
167 |
+
from .pipeline_onnx_stable_diffusion_inpaint import (
|
168 |
+
OnnxStableDiffusionInpaintPipeline,
|
169 |
+
)
|
170 |
+
from .pipeline_onnx_stable_diffusion_upscale import (
|
171 |
+
OnnxStableDiffusionUpscalePipeline,
|
172 |
+
)
|
173 |
+
|
174 |
+
try:
|
175 |
+
if not (is_transformers_available() and is_flax_available()):
|
176 |
+
raise OptionalDependencyNotAvailable()
|
177 |
+
except OptionalDependencyNotAvailable:
|
178 |
+
from ...utils.dummy_flax_objects import *
|
179 |
+
else:
|
180 |
+
from .pipeline_flax_stable_diffusion import FlaxStableDiffusionPipeline
|
181 |
+
from .pipeline_flax_stable_diffusion_img2img import (
|
182 |
+
FlaxStableDiffusionImg2ImgPipeline,
|
183 |
+
)
|
184 |
+
from .pipeline_flax_stable_diffusion_inpaint import (
|
185 |
+
FlaxStableDiffusionInpaintPipeline,
|
186 |
+
)
|
187 |
+
from .pipeline_output import FlaxStableDiffusionPipelineOutput
|
188 |
+
from .safety_checker_flax import FlaxStableDiffusionSafetyChecker
|
189 |
+
|
190 |
+
else:
|
191 |
+
import sys
|
192 |
+
|
193 |
+
sys.modules[__name__] = _LazyModule(
|
194 |
+
__name__,
|
195 |
+
globals()["__file__"],
|
196 |
+
_import_structure,
|
197 |
+
module_spec=__spec__,
|
198 |
+
)
|
199 |
+
|
200 |
+
for name, value in _dummy_objects.items():
|
201 |
+
setattr(sys.modules[__name__], name, value)
|
202 |
+
for name, value in _additional_imports.items():
|
203 |
+
setattr(sys.modules[__name__], name, value)
|