dvir-bria commited on
Commit
24f0751
1 Parent(s): 78fd408

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
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),