Update app.py

app.py

@@ -12,14 +12,26 @@ 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"
 )
 
 # device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe.to(device)
 
+# pipe.unet.load_attn_procs(
+# "./pytorch_custom_diffusion_weights.bin"
+# )
+
+# attn_path = "./pytorch_custom_diffusion_weights.bin"
+# state_dict = torch.load(attn_path, map_location="cpu")
+# pipe.unet.load_attn_procs(
+# state_dict
+# ) 
+
 pipe.unet.load_attn_procs(
-"./pytorch_custom_diffusion_weights.bin"
+    ".",
+    weight_name="pytorch_custom_diffusion_weights.bin"
 )
 
 def generate(prompt, steps, guidance):