Dhruv Diddi
commited on
Commit
·
9ddd3c5
1
Parent(s):
97a5a4e
feat: remove water mark processing
Browse files
model.py
CHANGED
@@ -269,7 +269,7 @@ class Model:
|
|
269 |
num_inference_steps=num_inference_steps_3,
|
270 |
generator=generator,
|
271 |
noise_level=100).images
|
272 |
-
self.apply_watermark_to_sd_x4_upscaler_results(out)
|
273 |
return out[0]
|
274 |
|
275 |
def run_stage2_3(
|
|
|
269 |
num_inference_steps=num_inference_steps_3,
|
270 |
generator=generator,
|
271 |
noise_level=100).images
|
272 |
+
# self.apply_watermark_to_sd_x4_upscaler_results(out)
|
273 |
return out[0]
|
274 |
|
275 |
def run_stage2_3(
|