Update app.py

app.py

@@ -527,9 +527,12 @@ def main():
     dtype = torch.float16
     
     vae = AutoencoderKL.from_pretrained("guangkaixu/GenPercept", subfolder='vae').to(dtype)
-    unet_depth_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/genpercept-depth', subfolder="unet").to(dtype)
-    unet_normal_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_normal_v1").to(dtype)
-    unet_dis_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_dis_v1").to(dtype)
+    unet_depth_v1 = UNet2DConditionModel.from_pretrained(
+                                            'guangkaixu/genpercept-depth', 
+                                            subfolder="unet",
+                                            use_safetensors=True).to(dtype)
+    unet_normal_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_normal_v1", use_safetensors=True).to(dtype)
+    unet_dis_v1 = UNet2DConditionModel.from_pretrained('guangkaixu/GenPercept', subfolder="unet_dis_v1", use_safetensors=True).to(dtype)
     
     empty_text_embed = torch.from_numpy(np.load("./empty_text_embed.npy")).to(device, dtype)[None] # [1, 77, 1024]