search

IDEA-Research / DreamWaltz

[NeurIPS 2023] Official implementation of the paper "DreamWaltz: Make a Scene with Complex 3D Animatable Avatars".
https://idea-research.github.io/DreamWaltz/
Other
180 stars 9 forks source link

Where do I get the pretrained ControlNet? #3

Closed lyk412 closed 11 months ago

lyk412 commented 1 year ago

Thanks for your great work! I met a problem when I want to download the pretrained model. I can not find the below models on the Hugging Face. 

MODEL_CARDS = {
    'pose': "fusing/stable-diffusion-v1-5-controlnet-openpose",
    'depth': "fusing/stable-diffusion-v1-5-controlnet-depth",
    'canny': "fusing/stable-diffusion-v1-5-controlnet-canny",
    'seg': "fusing/stable-diffusion-v1-5-controlnet-seg",
    'normal': "fusing/stable-diffusion-v1-5-controlnet-normal",
}
Yukun-Huang commented 1 year ago

Thanks for your comment! The pre-trained models of ControlNet have been moved to: https://huggingface.co/lllyasviel/sd-controlnet-openpose. You can download the weights there.

xmu-xiaoma666 commented 1 year ago

Thanks for your comment! The pre-trained models of ControlNet have been moved to: https://huggingface.co/lllyasviel/sd-controlnet-openpose. You can download the weights there.

Hello, How many gpu hours are needed to train your model on a single NVIDIA 3090?