AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

Add UniDiffuser classes to __init__ files, modify transformer block to support pre- and post-LN, add fast default tests, fix some bugs.

Browse Source
This commit is contained in:
Daniel Gu
2023-04-14 00:54:02 -07:00
parent 1d213def63
commit 434b255060
3 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/diffusers/__init__.py
View File

@@ -129,6 +129,7 @@ else:
IFInpaintingSuperResolutionPipeline,
IFPipeline,
IFSuperResolutionPipeline,
ImageTextPipelineOutput,
LDMTextToImagePipeline,
PaintByExamplePipeline,
SemanticStableDiffusionPipeline,

1
src/diffusers/pipelines/unidiffuser/pipeline_unidiffuser.py
View File

@@ -1116,6 +1116,7 @@ class UniDiffuserPipeline(DiffusionPipeline):
generator,
vae_latents,
)
# print(f"Image vae latent shape: {image_vae_latents.shape}")
# Prepare image CLIP latents
image_clip_latents = self.prepare_image_clip_latents(

4
tests/pipelines/unidiffuser/test_unidiffuser.py
View File

@@ -24,8 +24,8 @@ from diffusers import (
from diffusers.utils import floats_tensor, slow
from diffusers.utils.testing_utils import require_torch_gpu
from ...pipeline_params import TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS, TEXT_GUIDED_IMAGE_VARIATION_PARAMS
from ...test_pipelines_common import PipelineTesterMixin
from ..pipeline_params import TEXT_GUIDED_IMAGE_VARIATION_BATCH_PARAMS, TEXT_GUIDED_IMAGE_VARIATION_PARAMS
from ..test_pipelines_common import PipelineTesterMixin
class UniDiffuserPipelineFastTests(PipelineTesterMixin, unittest.TestCase):