Skip to content

Commit 29b2c93

Browse files
Make repo structure consistent (#1862)
* move files a bit * more refactors * fix more * more fixes * fix more onnx * make style * upload * fix * up * fix more * up again * up * small fix * Update src/diffusers/__init__.py Co-authored-by: Pedro Cuenca <[email protected]> * correct Co-authored-by: Pedro Cuenca <[email protected]>
1 parent ab0e92f commit 29b2c93

File tree

90 files changed

+1914
-1369
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

90 files changed

+1914
-1369
lines changed

docs/README.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -155,9 +155,9 @@ adds a link to its documentation with this syntax: \[\`XXXClass\`\] or \[\`funct
155155
function to be in the main package.
156156

157157
If you want to create a link to some internal class or function, you need to
158-
provide its path. For instance: \[\`pipeline_utils.ImagePipelineOutput\`\]. This will be converted into a link with
159-
`pipeline_utils.ImagePipelineOutput` in the description. To get rid of the path and only keep the name of the object you are
160-
linking to in the description, add a ~: \[\`~pipeline_utils.ImagePipelineOutput\`\] will generate a link with `ImagePipelineOutput` in the description.
158+
provide its path. For instance: \[\`pipelines.ImagePipelineOutput\`\]. This will be converted into a link with
159+
`pipelines.ImagePipelineOutput` in the description. To get rid of the path and only keep the name of the object you are
160+
linking to in the description, add a ~: \[\`~pipelines.ImagePipelineOutput\`\] will generate a link with `ImagePipelineOutput` in the description.
161161

162162
The same works for methods so you can either use \[\`XXXClass.method\`\] or \[~\`XXXClass.method\`\].
163163

docs/source/api/diffusion_pipeline.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,4 +39,4 @@ Any pipeline object can be saved locally with [`~DiffusionPipeline.save_pretrain
3939
## ImagePipelineOutput
4040
By default diffusion pipelines return an object of class
4141

42-
[[autodoc]] pipeline_utils.ImagePipelineOutput
42+
[[autodoc]] pipelines.ImagePipelineOutput

docs/source/api/models.mdx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,13 +41,13 @@ The models are built on the base class ['ModelMixin'] that is a `torch.nn.module
4141
[[autodoc]] models.vae.DecoderOutput
4242

4343
## VQEncoderOutput
44-
[[autodoc]] models.vae.VQEncoderOutput
44+
[[autodoc]] models.vq_model.VQEncoderOutput
4545

4646
## VQModel
4747
[[autodoc]] VQModel
4848

4949
## AutoencoderKLOutput
50-
[[autodoc]] models.vae.AutoencoderKLOutput
50+
[[autodoc]] models.autoencoder_kl.AutoencoderKLOutput
5151

5252
## AutoencoderKL
5353
[[autodoc]] AutoencoderKL

docs/source/api/outputs.mdx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ pipeline = DDIMPipeline.from_pretrained("google/ddpm-cifar10-32")
2525
outputs = pipeline()
2626
```
2727

28-
The `outputs` object is a [`~pipeline_utils.ImagePipelineOutput`], as we can see in the
28+
The `outputs` object is a [`~pipelines.ImagePipelineOutput`], as we can see in the
2929
documentation of that class below, it means it has an image attribute.
3030

3131
You can access each attribute as you would usually do, and if that attribute has not been returned by the model, you will get `None`:

examples/community/bit_diffusion.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,7 @@
22

33
import torch
44

5-
from diffusers import DDIMScheduler, DDPMScheduler, DiffusionPipeline, UNet2DConditionModel
6-
from diffusers.pipeline_utils import ImagePipelineOutput
5+
from diffusers import DDIMScheduler, DDPMScheduler, DiffusionPipeline, ImagePipelineOutput, UNet2DConditionModel
76
from diffusers.schedulers.scheduling_ddim import DDIMSchedulerOutput
87
from diffusers.schedulers.scheduling_ddpm import DDPMSchedulerOutput
98
from einops import rearrange, reduce

examples/community/checkpoint_merger.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,7 @@
55
import torch
66

77
from diffusers import DiffusionPipeline, __version__
8-
from diffusers.pipeline_utils import (
9-
CONFIG_NAME,
10-
DIFFUSERS_CACHE,
11-
ONNX_WEIGHTS_NAME,
12-
SCHEDULER_CONFIG_NAME,
13-
WEIGHTS_NAME,
14-
)
8+
from diffusers.utils import CONFIG_NAME, DIFFUSERS_CACHE, ONNX_WEIGHTS_NAME, SCHEDULER_CONFIG_NAME, WEIGHTS_NAME
159
from huggingface_hub import snapshot_download
1610

1711

examples/community/composable_stable_diffusion.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,21 +17,21 @@
1717

1818
import torch
1919

20-
from diffusers.utils import is_accelerate_available
21-
from packaging import version
22-
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
23-
24-
from ...configuration_utils import FrozenDict
25-
from ...models import AutoencoderKL, UNet2DConditionModel
26-
from ...pipeline_utils import DiffusionPipeline
27-
from ...schedulers import (
20+
from diffusers import DiffusionPipeline
21+
from diffusers.configuration_utils import FrozenDict
22+
from diffusers.models import AutoencoderKL, UNet2DConditionModel
23+
from diffusers.schedulers import (
2824
DDIMScheduler,
2925
DPMSolverMultistepScheduler,
3026
EulerAncestralDiscreteScheduler,
3127
EulerDiscreteScheduler,
3228
LMSDiscreteScheduler,
3329
PNDMScheduler,
3430
)
31+
from diffusers.utils import is_accelerate_available
32+
from packaging import version
33+
from transformers import CLIPFeatureExtractor, CLIPTextModel, CLIPTokenizer
34+
3535
from ...utils import deprecate, logging
3636
from . import StableDiffusionPipelineOutput
3737
from .safety_checker import StableDiffusionSafetyChecker

examples/community/imagic_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212

1313
import PIL
1414
from accelerate import Accelerator
15+
from diffusers import DiffusionPipeline
1516
from diffusers.models import AutoencoderKL, UNet2DConditionModel
16-
from diffusers.pipeline_utils import DiffusionPipeline
1717
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
1818
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
1919
from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler

examples/community/img2img_inpainting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,9 @@
55
import torch
66

77
import PIL
8+
from diffusers import DiffusionPipeline
89
from diffusers.configuration_utils import FrozenDict
910
from diffusers.models import AutoencoderKL, UNet2DConditionModel
10-
from diffusers.pipeline_utils import DiffusionPipeline
1111
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
1212
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
1313
from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler

examples/community/interpolate_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,9 @@
66
import numpy as np
77
import torch
88

9+
from diffusers import DiffusionPipeline
910
from diffusers.configuration_utils import FrozenDict
1011
from diffusers.models import AutoencoderKL, UNet2DConditionModel
11-
from diffusers.pipeline_utils import DiffusionPipeline
1212
from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput
1313
from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
1414
from diffusers.schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler

0 commit comments

Comments
 (0)