From 57f88c687b4724d5f2ab139c2952952208784820 Mon Sep 17 00:00:00 2001
From: Justin John <34035011+justinjohn0306@users.noreply.github.com>
Date: Thu, 16 Nov 2023 20:40:15 +0530
Subject: [PATCH] enable auto downloading depth and openpose model

---
 animation/Rerender_A_Video/install.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/animation/Rerender_A_Video/install.py b/animation/Rerender_A_Video/install.py
index 9ae6f29..87f34e2 100644
--- a/animation/Rerender_A_Video/install.py
+++ b/animation/Rerender_A_Video/install.py
@@ -64,6 +64,16 @@ def download_controlnet_hed():
            'resolve/main/models/control_sd15_hed.pth')
     download(url, 'models')
 
+def download_controlnet_depth():
+    url = ('https://huggingface.co/lllyasviel/ControlNet/'
+           'resolve/main/models/control_sd15_depth.pth')
+    download(url, 'models')
+
+def download_controlnet_openpose():
+    url = ('https://huggingface.co/lllyasviel/ControlNet/'
+           'resolve/main/models/control_sd15_openpose.pth')
+    download(url, 'models')
+
 
 def download_vae():
     url = ('https://huggingface.co/stabilityai/sd-vae-ft-mse-original'
@@ -75,4 +85,6 @@ build_ebsynth()
 download_gmflow_ckpt()
 download_controlnet_canny()
 download_controlnet_hed()
+download_controlnet_depth()
+download_controlnet_openpose()
 download_vae()