Update app.py

app.py

@@ -5,11 +5,17 @@ import torch
 import torch.cuda.amp as amp
 from diffusers import StableDiffusionPipeline
 
-modelid = "CompVis/stable-diffusion-v1-4"
+
+
+modelid = "stabilityai/stable-diffusion-2-1"
+
 device = torch.device("cpu")  # Default to CPU device
 if torch.cuda.is_available():
     device = torch.device("cuda")
-pipe = StableDiffusionPipeline.from_pretrained(modelid, use_auth_token=auth_token)
+
+# Use the DPMSolverMultistepScheduler (DPM-Solver++) scheduler here instead
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.to(device)
 
 def generate(prompt):