Update app.py

app.py

@@ -70,14 +70,11 @@ if is_colab:
 
 else:
   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()
+      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")
 
 device = "GPU 🔥" if torch.cuda.is_available() else "CPU 🥶"