williamberman commited on
Commit
a6167b6
1 Parent(s): 3722f7e

enable queue

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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)