Update app.py

app.py

@@ -11,14 +11,22 @@ dtype = torch.float16 if device == "cuda" else torch.float32
 
 pipe = DiffusionPipeline.from_pretrained(
     "stable-diffusion-v1-5/stable-diffusion-v1-5",
-    torch_dtype=torch.bfloat16,
-    cache_dir="/tmp/huggingface"
+    torch_dtype=torch.float16 if device=="cuda" else torch.float32,
+    cache_dir="/tmp/huggingface",
+    use_safetensors=True,
+    safety_checker=None
 )
 
+# pipe.safety_checker = None
+
 # device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe.to(device)
 
+pipe.enable_attention_slicing()
+pipe.enable_model_cpu_offload()
+
+
 # pipe.unet.load_attn_procs(
 # "./pytorch_custom_diffusion_weights.bin"
 # )