fix setup for watermarking.

Author: sayakpaul

setup.py

@@ -117,6 +117,7 @@
     "torchvision",
     "transformers>=4.25.1",
     "urllib3<=2.0.0",
+    "invisible-watermark"
 ]
 
 # this is a lookup table with items like:
@@ -207,6 +208,7 @@ def run(self):
     "scipy",
     "torchvision",
     "transformers",
+    "invisible-watermark",
 )
 extras["torch"] = deps_list("torch", "accelerate")
 

src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py

@@ -31,13 +31,19 @@
 from ...utils import (
     is_accelerate_available,
     is_accelerate_version,
+    is_invisible_watermark_available,
     logging,
     randn_tensor,
     replace_example_docstring,
 )
 from ..pipeline_utils import DiffusionPipeline
 from . import StableDiffusionXLPipelineOutput
-from .watermark import StableDiffusionXLWatermarker
+
+
+if is_invisible_watermark_available():
+    from .watermark import StableDiffusionXLWatermarker
+else:
+    raise ImportError("StableDiffusionXLWatermarker requires the `invisible-watermark` library installed.")
 
 
 logger = logging.get_logger(__name__)  # pylint: disable=invalid-name

tests/others/test_dependencies.py

@@ -34,4 +34,6 @@ def test_backend_registration(self):
                 for backend in cls_module._backends:
                     if backend == "k_diffusion":
                         backend = "k-diffusion"
+                    elif backend == "invisible_watermark":
+                        backend = "invisible-watermark"
                     assert backend in deps, f"{backend} is not in the deps table!"