Update app.py

app.py

@@ -46,15 +46,16 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
  pip = pip_2_1
  else:
  pip = pip_1_4
- # register_free_upblock2d(pip, b1=1.0, b2=1.0, s1=1.0, s2=1.0)
- # register_free_crossattn_upblock2d(pip, b1=1.0, b2=1.0, s1=1.0, s2=1.0)
+ 
+ register_free_upblock2d(pip, b1=1.0, b2=1.0, s1=1.0, s2=1.0)
+ register_free_crossattn_upblock2d(pip, b1=1.0, b2=1.0, s1=1.0, s2=1.0)
  
  torch.manual_seed(seed)
  print("Generating SD:")
  sd_image = pip(prompt, num_inference_steps=25).images[0] 
 
- # register_free_upblock2d(pip, b1=b1, b2=b2, s1=s1, s2=s1)
- # register_free_crossattn_upblock2d(pip, b1=b1, b2=b2, s1=s1, s2=s1)
+ register_free_upblock2d(pip, b1=b1, b2=b2, s1=s1, s2=s1)
+ register_free_crossattn_upblock2d(pip, b1=b1, b2=b2, s1=s1, s2=s1)
 
  torch.manual_seed(seed)
  print("Generating FreeU:")