upload

app.py

@@ -272,11 +272,11 @@ def main():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     
     vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae')
-    unet = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet")
+    unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_depth_v1")
     empty_text_embed = torch.from_numpy(np.load("./empty_text_embed.npy")).to(device, dtype)[None] # [1, 77, 1024]
     
     pipe = GenPerceptPipeline(vae=vae,
-                              unet=unet,
+                              unet=unet_depth_v1,
                               empty_text_embed=empty_text_embed)
     try:
         import xformers