williamberman
commited on
Commit
•
a6167b6
1
Parent(s):
3722f7e
enable queue
Browse files
app.py
CHANGED
@@ -7,10 +7,12 @@ from share_btn import community_icon_html, loading_icon_html, share_js
|
|
7 |
from sdxl import sdxl_diffusion_loop
|
8 |
from sdxl_models import SDXLUNet, SDXLVae, SDXLControlNetPreEncodedControlnetCond
|
9 |
import torchvision.transforms.functional as TF
|
10 |
-
from diffusion import make_sigmas
|
11 |
from huggingface_hub import hf_hub_download
|
12 |
import gc
|
13 |
|
|
|
|
|
14 |
pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16")
|
15 |
pipe.text_encoder.to("cuda")
|
16 |
pipe.text_encoder_2.to("cuda")
|
@@ -192,4 +194,4 @@ with image_blocks as demo:
|
|
192 |
"""
|
193 |
)
|
194 |
|
195 |
-
image_blocks.queue(max_size=25).launch()
|
|
|
7 |
from sdxl import sdxl_diffusion_loop
|
8 |
from sdxl_models import SDXLUNet, SDXLVae, SDXLControlNetPreEncodedControlnetCond
|
9 |
import torchvision.transforms.functional as TF
|
10 |
+
from diffusion import make_sigmas, set_with_tqdm
|
11 |
from huggingface_hub import hf_hub_download
|
12 |
import gc
|
13 |
|
14 |
+
set_with_tqdm(True)
|
15 |
+
|
16 |
pipe = AutoPipelineForInpainting.from_pretrained("diffusers/stable-diffusion-xl-1.0-inpainting-0.1", torch_dtype=torch.float16, variant="fp16")
|
17 |
pipe.text_encoder.to("cuda")
|
18 |
pipe.text_encoder_2.to("cuda")
|
|
|
194 |
"""
|
195 |
)
|
196 |
|
197 |
+
image_blocks.queue(max_size=25).launch(enable_queue=True)
|