Update app.py

app.py

@@ -68,12 +68,15 @@ if is_colab:
 
 
 else:
-  pipe = StableDiffusionPipeline.from_pretrained(
-      model_id,
-      revision="fp16" if torch.cuda.is_available() else "fp32",
-      torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
-      scheduler=scheduler
-    ).to("cuda")
+    pipe = StableDiffusionPipeline.from_pretrained(
+      current_model.path,
+      torch_dtype=torch.float16,
+      scheduler=DPMSolverMultistepScheduler.from_pretrained(current_model.path, subfolder="scheduler")
+      )
+    
+if torch.cuda.is_available():
+  pipe = pipe.to("cuda")
+  pipe.enable_xformers_memory_efficient_attention()
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"