Spaces:
Running
on
Zero
Running
on
Zero
alfredplpl
commited on
Commit
·
60d9680
1
Parent(s):
f55f6bf
Update app.py
Browse files
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,
|