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(
|