Closed G-force78 closed 1 year ago
I cant find the instructions on the readme, neither can I see where the depth model ends up.
you can download from here: https://huggingface.co/wf-genius/controlavideo-hed/blob/main/hed-network.pth
Yes have seen that thanks, I have now found out the path needs to be changed in inference.py to the working directory or wherever you downloaded it to. eg
_``` if control_mode == 'depth': annotator_model = DPTForDepthEstimation.from_pretrained("Intel/dpt-hybrid-midas").to("cuda") elif control_mode == 'canny': annotator_model = None elif control_mode == 'hed':
annotator_model = HEDNetwork('/content/control-a-video/model/annotator/hed-network.pth').to("cuda")
I cant find the instructions on the readme, neither can I see where the depth model ends up.