Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL
|
2 |
from diffusers.utils import load_image
|
3 |
from PIL import Image
|
4 |
import torch
|
@@ -20,6 +20,13 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
|
20 |
low_cpu_mem_usage=True,
|
21 |
offload_state_dict=True,
|
22 |
).to('cuda')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
23 |
# pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
|
24 |
pipe.enable_xformers_memory_efficient_attention()
|
25 |
pipe.force_zeros_for_empty_prompt = False
|
|
|
1 |
+
from diffusers import ControlNetModel, StableDiffusionXLControlNetPipeline, AutoencoderKL, EulerAncestralDiscreteScheduler
|
2 |
from diffusers.utils import load_image
|
3 |
from PIL import Image
|
4 |
import torch
|
|
|
20 |
low_cpu_mem_usage=True,
|
21 |
offload_state_dict=True,
|
22 |
).to('cuda')
|
23 |
+
pipe.scheduler = EulerAncestralDiscreteScheduler(
|
24 |
+
beta_start=0.00085,
|
25 |
+
beta_end=0.012,
|
26 |
+
beta_schedule="scaled_linear",
|
27 |
+
num_train_timesteps=1000,
|
28 |
+
steps_offset=1
|
29 |
+
)
|
30 |
# pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
|
31 |
pipe.enable_xformers_memory_efficient_attention()
|
32 |
pipe.force_zeros_for_empty_prompt = False
|