Skip to content

Commit 5b186b7

Browse files
authored
[Refactor] move ldm3d out of stable_diffusion. (#6263)
ldm3d.
1 parent ab0459f commit 5b186b7

File tree

5 files changed

+53
-6
lines changed

5 files changed

+53
-6
lines changed

docs/source/en/api/pipelines/stable_diffusion/ldm3d_diffusion.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,14 +31,14 @@ Make sure to check out the Stable Diffusion [Tips](overview#tips) section to lea
3131

3232
## StableDiffusionLDM3DPipeline
3333

34-
[[autodoc]] pipelines.stable_diffusion.pipeline_stable_diffusion_ldm3d.StableDiffusionLDM3DPipeline
34+
[[autodoc]] pipelines.stable_diffusion_ldm3d.pipeline_stable_diffusion_ldm3d.StableDiffusionLDM3DPipeline
3535
- all
3636
- __call__
3737

3838

3939
## LDM3DPipelineOutput
4040

41-
[[autodoc]] pipelines.stable_diffusion.pipeline_stable_diffusion_ldm3d.LDM3DPipelineOutput
41+
[[autodoc]] pipelines.stable_diffusion_ldm3d.pipeline_stable_diffusion_ldm3d.LDM3DPipelineOutput
4242
- all
4343
- __call__
4444

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@
209209
]
210210
)
211211
_import_structure["stable_diffusion_diffedit"] = ["StableDiffusionDiffEditPipeline"]
212+
_import_structure["stable_diffusion_ldm3d"] = ["StableDiffusionLDM3DPipeline"]
212213
_import_structure["stable_diffusion_panorama"] = ["StableDiffusionPanoramaPipeline"]
213214
_import_structure["t2i_adapter"] = [
214215
"StableDiffusionAdapterPipeline",
@@ -427,7 +428,6 @@
427428
StableDiffusionInpaintPipeline,
428429
StableDiffusionInstructPix2PixPipeline,
429430
StableDiffusionLatentUpscalePipeline,
430-
StableDiffusionLDM3DPipeline,
431431
StableDiffusionPipeline,
432432
StableDiffusionUpscalePipeline,
433433
StableUnCLIPImg2ImgPipeline,
@@ -436,6 +436,7 @@
436436
from .stable_diffusion_attend_and_excite import StableDiffusionAttendAndExcitePipeline
437437
from .stable_diffusion_diffedit import StableDiffusionDiffEditPipeline
438438
from .stable_diffusion_gligen import StableDiffusionGLIGENPipeline, StableDiffusionGLIGENTextImagePipeline
439+
from .stable_diffusion_ldm3d import StableDiffusionLDM3DPipeline
439440
from .stable_diffusion_panorama import StableDiffusionPanoramaPipeline
440441
from .stable_diffusion_safe import StableDiffusionPipelineSafe
441442
from .stable_diffusion_sag import StableDiffusionSAGPipeline

src/diffusers/pipelines/stable_diffusion/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,6 @@
4040
_import_structure["pipeline_stable_diffusion_inpaint_legacy"] = ["StableDiffusionInpaintPipelineLegacy"]
4141
_import_structure["pipeline_stable_diffusion_instruct_pix2pix"] = ["StableDiffusionInstructPix2PixPipeline"]
4242
_import_structure["pipeline_stable_diffusion_latent_upscale"] = ["StableDiffusionLatentUpscalePipeline"]
43-
_import_structure["pipeline_stable_diffusion_ldm3d"] = ["StableDiffusionLDM3DPipeline"]
4443
_import_structure["pipeline_stable_diffusion_model_editing"] = ["StableDiffusionModelEditingPipeline"]
4544
_import_structure["pipeline_stable_diffusion_paradigms"] = ["StableDiffusionParadigmsPipeline"]
4645
_import_structure["pipeline_stable_diffusion_upscale"] = ["StableDiffusionUpscalePipeline"]
@@ -124,7 +123,6 @@
124123
from .pipeline_stable_diffusion_latent_upscale import (
125124
StableDiffusionLatentUpscalePipeline,
126125
)
127-
from .pipeline_stable_diffusion_ldm3d import StableDiffusionLDM3DPipeline
128126
from .pipeline_stable_diffusion_upscale import StableDiffusionUpscalePipeline
129127
from .pipeline_stable_unclip import StableUnCLIPPipeline
130128
from .pipeline_stable_unclip_img2img import StableUnCLIPImg2ImgPipeline
Lines changed: 48 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,48 @@
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_stable_diffusion_ldm3d"] = ["StableDiffusionLDM3DPipeline"]
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+
32+
except OptionalDependencyNotAvailable:
33+
from ...utils.dummy_torch_and_transformers_objects import *
34+
else:
35+
from .pipeline_stable_diffusion_ldm3d import StableDiffusionLDM3DPipeline
36+
37+
else:
38+
import sys
39+
40+
sys.modules[__name__] = _LazyModule(
41+
__name__,
42+
globals()["__file__"],
43+
_import_structure,
44+
module_spec=__spec__,
45+
)
46+
47+
for name, value in _dummy_objects.items():
48+
setattr(sys.modules[__name__], name, value)

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_ldm3d.py renamed to src/diffusers/pipelines/stable_diffusion_ldm3d/pipeline_stable_diffusion_ldm3d.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
)
3838
from ...utils.torch_utils import randn_tensor
3939
from ..pipeline_utils import DiffusionPipeline
40-
from .safety_checker import StableDiffusionSafetyChecker
40+
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
4141

4242

4343
logger = logging.get_logger(__name__) # pylint: disable=invalid-name

0 commit comments

Comments
 (0)