Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -18,6 +18,7 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
|
18 |
controlnet=controlnet,
|
19 |
torch_dtype=torch.float16,
|
20 |
).to('cuda')
|
|
|
21 |
pipe.enable_xformers_memory_efficient_attention()
|
22 |
|
23 |
low_threshold = 100
|
@@ -51,8 +52,6 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
|
|
51 |
input_image = resize_image(input_image)
|
52 |
|
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),
|
|
|
18 |
controlnet=controlnet,
|
19 |
torch_dtype=torch.float16,
|
20 |
).to('cuda')
|
21 |
+
pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
|
22 |
pipe.enable_xformers_memory_efficient_attention()
|
23 |
|
24 |
low_threshold = 100
|
|
|
52 |
input_image = resize_image(input_image)
|
53 |
|
54 |
canny_image = get_canny_filter(input_image)
|
|
|
|
|
55 |
|
56 |
images = pipe(
|
57 |
prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),
|