fix path

liuwenran · Dec 8, 2023 · a555560 · a555560
1 parent b9afc33
commit a555560
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/projects/powerpaint/gradio_PowerPaint.py b/projects/powerpaint/gradio_PowerPaint.py
@@ -34,9 +34,9 @@
     initialize_tokens=['a', 'a', 'a'],
     num_vectors_per_token=10)
 pipe.unet.load_state_dict(
-    torch.load('./models/diffusion_pytorch_model.bin'), strict=False)
+    torch.load('./models/unet/diffusion_pytorch_model.bin'), strict=False)
 pipe.text_encoder.load_state_dict(
-    torch.load('./models/pytorch_model.bin'), strict=False)
+    torch.load('./models/text_encoder/pytorch_model.bin'), strict=False)
 pipe = pipe.to('cuda')
 
 depth_estimator = DPTForDepthEstimation.from_pretrained(