Update app.py

app.py

@@ -16,9 +16,9 @@ llm = pipeline(
 )
 
 # Step 2: 加载 Stable Diffusion 模型
+# 移除无效的 revision 参数，仅使用 torch_dtype 加速加载
 sd_v15 = StableDiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", 
-    revision="fp16", 
+    "runwayml/stable-diffusion-v1-5",
     torch_dtype=torch.float16
 )
 sd_v15 = sd_v15.to("cuda" if torch.cuda.is_available() else "cpu")
@@ -107,3 +107,4 @@ with gr.Blocks(title="Prompt-to-Image Generator") as demo:
 # Step 4: 启动应用
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)
+