Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
11 |
"briaai/ControlNet-Canny",
|
12 |
torch_dtype=torch.float16
|
13 |
)
|
|
|
14 |
|
15 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
16 |
"briaai/BRIA-2.0",
|
@@ -52,17 +53,11 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
|
|
52 |
canny_image = get_canny_filter(input_image)
|
53 |
|
54 |
pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
).images
|
61 |
-
else:
|
62 |
-
images = pipe(
|
63 |
-
prompt, force_zeros_for_empty_prompt=False, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
64 |
-
generator=generator,
|
65 |
-
).images
|
66 |
|
67 |
return [canny_image,images[0]]
|
68 |
|
|
|
11 |
"briaai/ControlNet-Canny",
|
12 |
torch_dtype=torch.float16
|
13 |
)
|
14 |
+
# force_zeros_for_empty_prompt=False
|
15 |
|
16 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
17 |
"briaai/BRIA-2.0",
|
|
|
53 |
canny_image = get_canny_filter(input_image)
|
54 |
|
55 |
pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
|
56 |
+
|
57 |
+
images = pipe(
|
58 |
+
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|
59 |
+
generator=generator,
|
60 |
+
).images
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
|
62 |
return [canny_image,images[0]]
|
63 |
|