alfredplpl commited on
Commit
60d9680
1 Parent(s): f55f6bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ pipe = StableDiffusionPipeline.from_pretrained(
16
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
17
  scheduler=scheduler)
18
  pipe.enable_xformers_memory_efficient_attention()
19
-
 
20
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
21
  model_id,
22
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
 
16
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
17
  scheduler=scheduler)
18
  pipe.enable_xformers_memory_efficient_attention()
19
+ pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.4, b2=1.6)
20
+ # b1: 1.4, b2: 1.6, s1: 0.9, s2: 0.2
21
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
22
  model_id,
23
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,