Spaces:
Runtime error
Runtime error
Commit
·
85a7b79
1
Parent(s):
6fac16d
Counteract washed-out/gray generation
Browse filesThis might be caused by the watermark? Further investigation needed
app.py
CHANGED
@@ -109,7 +109,7 @@ def detect(image):
|
|
109 |
return max(0.0, 1-1/math.log(5/p_value,10))
|
110 |
|
111 |
def generate(prompt):
|
112 |
-
return pipe(prompt=prompt, num_inference_steps=25, latents=get_noise()).images[0]
|
113 |
|
114 |
# optimize for speed
|
115 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
|
|
109 |
return max(0.0, 1-1/math.log(5/p_value,10))
|
110 |
|
111 |
def generate(prompt):
|
112 |
+
return pipe(prompt=prompt, negative_prompt="monochrome", num_inference_steps=25, latents=get_noise()).images[0]
|
113 |
|
114 |
# optimize for speed
|
115 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|