update space

app.py

@@ -22,8 +22,11 @@ else:
     )
     pipe = pipe.to(device)
 
+    # Quantize the model
+    pipe.unet = torch.quantization.convert(pipe.unet, inplace=True)
+
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 256
+MAX_IMAGE_SIZE = 512
 
 
 def infer(
@@ -123,7 +126,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=256,
+                    value=512,
                 )
 
                 height = gr.Slider(
@@ -131,7 +134,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=256,
+                    value=512,
                 )
 
             with gr.Row():
@@ -169,4 +172,4 @@ with gr.Blocks(css=css) as demo:
         outputs=[result],
     )
 
-demo.queue().launch()
+demo.queue().launch(share=True)