Spaces:
Sleeping
Sleeping
Vivien Chappelier
commited on
Commit
•
d5025f0
1
Parent(s):
0c16ea5
is default watermarking enabled?
Browse files
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
|