Vivien Chappelier commited on
Commit
d5025f0
1 Parent(s): 0c16ea5

is default watermarking enabled?

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -27,6 +27,10 @@ class BZHStableSignatureDemo(object):
27
  def __init__(self, *args, **kwargs):
28
  super().__init__(*args, **kwargs)
29
  self.pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16").to("cuda")
 
 
 
 
30
 
31
  # load the patched VQ-VAEs
32
  sd1 = deepcopy(self.pipe.vae.state_dict()) # save initial state dict
 
27
  def __init__(self, *args, **kwargs):
28
  super().__init__(*args, **kwargs)
29
  self.pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16").to("cuda")
30
+ try:
31
+ print("self.pipe.watermark = ", self.pipe.watermark)
32
+ except:
33
+ print("no self.pipe.watermark")
34
 
35
  # load the patched VQ-VAEs
36
  sd1 = deepcopy(self.pipe.vae.state_dict()) # save initial state dict