Update app.py

app.py

@@ -1,4 +1,3 @@
-import os
 import random
 import uuid
 import json
@@ -13,9 +12,9 @@ if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
-USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
-ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
+MAX_IMAGE_SIZE = 4096
+USE_TORCH_COMPILE = "1"
+ENABLE_CPU_OFFLOAD = "1"
 
 if torch.cuda.is_available():
     pipe = StableDiffusionXLPipeline.from_pretrained(
@@ -72,6 +71,7 @@ footer {
     visibility: hidden
 }
 '''
+    
 with gr.Blocks(css=css) as demo:
     gr.Markdown("""# 梦羽的模型生成器
         ### 快速生成NoobXL的模型图片.""")