mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
update
This commit is contained in:
@@ -40,7 +40,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from .modeling_roberta_series import RobertaSeriesModelWithTransformation
|
||||
from .pipeline_output import AltDiffusionPipelineOutput
|
||||
|
||||
|
||||
@@ -43,7 +43,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from .modeling_roberta_series import RobertaSeriesModelWithTransformation
|
||||
from .pipeline_output import AltDiffusionPipelineOutput
|
||||
|
||||
|
||||
@@ -27,6 +27,8 @@ from ....loaders import FromSingleFileMixin, StableDiffusionLoraLoaderMixin, Tex
|
||||
from ....models import AutoencoderKL, ControlNetXSAdapter, UNet2DConditionModel, UNetControlNetXSModel
|
||||
from ....models.lora import adjust_lora_scale_text_encoder
|
||||
from ....schedulers import KarrasDiffusionSchedulers
|
||||
from ....stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ....stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ....utils import (
|
||||
USE_PEFT_BACKEND,
|
||||
deprecate,
|
||||
@@ -38,8 +40,6 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import is_compiled_module, is_torch_version, randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -38,6 +38,7 @@ from ....models.attention_processor import (
|
||||
)
|
||||
from ....models.lora import adjust_lora_scale_text_encoder
|
||||
from ....schedulers import KarrasDiffusionSchedulers
|
||||
from ....stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
||||
from ....utils import (
|
||||
USE_PEFT_BACKEND,
|
||||
logging,
|
||||
@@ -47,11 +48,10 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import is_compiled_module, is_torch_version, randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline
|
||||
from ...stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
||||
|
||||
|
||||
if is_invisible_watermark_available():
|
||||
from ..stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
|
||||
from ...stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
|
||||
|
||||
|
||||
from ....utils import is_torch_xla_available
|
||||
|
||||
@@ -26,8 +26,8 @@ from ....schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler
|
||||
from ....utils import deprecate, is_torch_xla_available, logging
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from .image_encoder import PaintByExampleImageEncoder
|
||||
|
||||
|
||||
|
||||
@@ -8,10 +8,10 @@ from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer
|
||||
from ....image_processor import VaeImageProcessor
|
||||
from ....models import AutoencoderKL, UNet2DConditionModel
|
||||
from ....schedulers import KarrasDiffusionSchedulers
|
||||
from ....stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ....utils import deprecate, is_torch_xla_available, logging
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from .pipeline_output import SemanticStableDiffusionPipelineOutput
|
||||
|
||||
|
||||
|
||||
@@ -38,8 +38,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -41,8 +41,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -37,8 +37,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -42,9 +42,9 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.clip_image_project_model import CLIPImageProjection
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.clip_image_project_model import CLIPImageProjection
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -43,7 +43,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput, StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -49,7 +49,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
||||
from ...stable_diffusion_xl.pipeline_output import StableDiffusionXLPipelineOutput
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -43,7 +43,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -34,8 +34,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -35,8 +35,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -30,8 +30,8 @@ from ....schedulers import DDIMScheduler
|
||||
from ....utils import PIL_INTERPOLATION, USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline
|
||||
from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -11,7 +11,7 @@ from ....schedulers import DDIMScheduler, LMSDiscreteScheduler, PNDMScheduler
|
||||
from ....utils import deprecate, logging
|
||||
from ...onnx_utils import ORT_TO_NP_TYPE, OnnxRuntimeModel
|
||||
from ...pipeline_utils import DiffusionPipeline
|
||||
from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -30,8 +30,8 @@ from ....schedulers import KarrasDiffusionSchedulers
|
||||
from ....utils import PIL_INTERPOLATION, USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline
|
||||
from ..stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__)
|
||||
|
||||
@@ -27,8 +27,8 @@ from ....schedulers.scheduling_utils import SchedulerMixin
|
||||
from ....utils import USE_PEFT_BACKEND, deprecate, logging, scale_lora_layers, unscale_lora_layers
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -33,8 +33,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -47,8 +47,8 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ..stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
|
||||
from ...stable_diffusion.safety_checker import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
logger = logging.get_logger(__name__) # pylint: disable=invalid-name
|
||||
|
||||
@@ -25,7 +25,7 @@ from ....utils import (
|
||||
)
|
||||
from ....utils.torch_utils import randn_tensor
|
||||
from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
from ..stable_diffusion import StableDiffusionSafetyChecker
|
||||
from ...stable_diffusion import StableDiffusionSafetyChecker
|
||||
|
||||
|
||||
if is_torch_xla_available():
|
||||
|
||||
@@ -39,7 +39,7 @@ from ...pipeline_utils import DiffusionPipeline, StableDiffusionMixin
|
||||
|
||||
|
||||
if is_invisible_watermark_available():
|
||||
from ..stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
|
||||
from ...stable_diffusion_xl.watermark import StableDiffusionXLWatermarker
|
||||
|
||||
|
||||
from ....utils import is_torch_xla_available
|
||||
|
||||
@@ -26,6 +26,7 @@ from diffusers.pipelines.deprecated.blip_diffusion.blip_image_processing import
|
||||
from diffusers.pipelines.deprecated.blip_diffusion.modeling_blip2 import Blip2QFormerModel
|
||||
from diffusers.pipelines.deprecated.blip_diffusion.modeling_ctx_clip import ContextCLIPTextModel
|
||||
from diffusers.utils.testing_utils import enable_full_determinism
|
||||
|
||||
from ..test_pipelines_common import PipelineTesterMixin
|
||||
|
||||
|
||||
@@ -194,9 +195,9 @@ class BlipDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
|
||||
[0.5329548, 0.8372512, 0.33269387, 0.82096875, 0.43657133, 0.3783, 0.5953028, 0.51934963, 0.42142007]
|
||||
)
|
||||
|
||||
assert (
|
||||
np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
|
||||
), f" expected_slice {image_slice.flatten()}, but got {image_slice.flatten()}"
|
||||
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2, (
|
||||
f" expected_slice {image_slice.flatten()}, but got {image_slice.flatten()}"
|
||||
)
|
||||
|
||||
@unittest.skip("Test not supported because of complexities in deriving query_embeds.")
|
||||
def test_encode_prompt_works_in_isolation(self):
|
||||
|
||||
Reference in New Issue
Block a user