diff --git a/run.py b/run.py index bb45cd57..e34d4250 100644 --- a/run.py +++ b/run.py @@ -6,6 +6,9 @@ sys.path.insert(0, os.getcwd()) # must come before ANY torch or fastai imports # import toolkit.cuda_malloc +# turn off diffusers telemetry until I can figure out how to make it opt-in +os.environ['DISABLE_TELEMETRY'] = 'YES' + # check if we have DEBUG_TOOLKIT in env if os.environ.get("DEBUG_TOOLKIT", "0") == "1": # set torch to trace mode diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py index 9a827f82..98a67e7e 100644 --- a/toolkit/stable_diffusion_model.py +++ b/toolkit/stable_diffusion_model.py @@ -310,6 +310,9 @@ class StableDiffusion: else: Pipe = StableDiffusionAdapterPipeline extra_args['adapter'] = self.adapter + else: + if self.is_xl: + extra_args['add_watermarker'] = False # TODO add clip skip if self.is_xl: @@ -321,7 +324,6 @@ class StableDiffusion: tokenizer=self.tokenizer[0], tokenizer_2=self.tokenizer[1], scheduler=noise_scheduler, - add_watermarker=False, **extra_args ).to(self.device_torch) # force turn that (ruin your images with obvious green and red dots) the #$@@ off!!!