Update app.py

app.py

@@ -4,16 +4,13 @@ import numpy as np
 import spaces
 import torch
 import os
-from diffusers import StableDiffusionXLPipeline, AutoencoderKL
 from huggingface_hub import hf_hub_download
+from diffusers import StableDiffusionXLPipeline, AutoencoderKL, EulerAncestralDiscreteScheduler
 from compel import Compel, ReturnedEmbeddingsType
 
-import re
-
 # =====================================
 # Prompt weights
 # =====================================
-import torch
 import re
 def parse_prompt_attention(text):
     re_attention = re.compile(r"""
@@ -224,7 +221,7 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是此项目只支持GPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 1536
+MAX_IMAGE_SIZE = 2048
 
 if torch.cuda.is_available():
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -240,6 +237,8 @@ if torch.cuda.is_available():
         use_safetensors=True,
         torch_dtype=torch.float16,
     )
+    pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+    pipe.enable_xformers_memory_efficient_attention()
     pipe.to("cuda")
 
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
@@ -398,4 +397,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)