Spaces:
Running
on
Zero
Running
on
Zero
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,
|