Update app.py

app.py

@@ -89,8 +89,8 @@ def infer(prompt, video_path, condition, video_length, is_long_video):
     tokenizer = CLIPTokenizer.from_pretrained(sd_path, subfolder="tokenizer")
     text_encoder = CLIPTextModel.from_pretrained(sd_path, subfolder="text_encoder").to(dtype=torch.float16)
     vae = AutoencoderKL.from_pretrained(sd_path, subfolder="vae").to(dtype=torch.float16)
-    unet = UNet3DConditionModel.from_pretrained_2d(sd_path, subfolder="unet").to(dtype=torch.float16)
-    controlnet = ControlNetModel3D.from_pretrained_2d(controlnet_dict[condition]).to(dtype=torch.float16)
+    unet = UNet3DConditionModel.from_pretrained(sd_path, subfolder="unet").to(dtype=torch.float16)
+    controlnet = ControlNetModel3D.from_pretrained(controlnet_dict[condition]).to(dtype=torch.float16)
     interpolater = IFNet(ckpt_path=inter_path).to(dtype=torch.float16)
     scheduler=DDIMScheduler.from_pretrained(sd_path, subfolder="scheduler")