diff --git a/__init__.py b/__init__.py index 59bbb1d..9d9a494 100644 --- a/__init__.py +++ b/__init__.py @@ -130,8 +130,8 @@ from .util_nodes import PreViewVideo, LoadVideo,CombineVideo NODE_CLASS_MAPPINGS = { "CombineVideo":CombineVideo, - "LoadVideo":LoadVideo, "PreViewVideo":PreViewVideo, "DHLiveNode": DHLiveNode, - "StaticVideo":StaticVideo + "StaticVideo":StaticVideo, + "DHLIVELoadVideo":LoadVideo, } \ No newline at end of file diff --git a/checkpoint/render.pth b/checkpoint/render.pth deleted file mode 100644 index a1dbc44..0000000 Binary files a/checkpoint/render.pth and /dev/null differ diff --git a/checkpoint/render.pth b/checkpoint/render.pth new file mode 120000 index 0000000..01a0336 --- /dev/null +++ b/checkpoint/render.pth @@ -0,0 +1 @@ +/.xgcos/.links/c96bf0f1-71a4-5e76-a469-5444639bf8b2 \ No newline at end of file diff --git a/talkingface/render_model.py b/talkingface/render_model.py index e394cfd..52195df 100644 --- a/talkingface/render_model.py +++ b/talkingface/render_model.py @@ -154,10 +154,10 @@ class RenderModel: self.upscale_net.load_state_dict(checkpoint) self.upscale_net.eval() elif model_name == "CodeFormer": - pretrain_model_url = 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer_inpainting.pth' + pretrain_model_url = 'https://github.com/sczhou/CodeFormer/releases/download/v0.1.0/codeformer.pth' if self.model_name != model_name: - self.upscale_net = ARCH_REGISTRY.get('CodeFormer')(dim_embd=512, codebook_size=512, n_head=8, n_layers=9, - connect_list=['32', '64', '128']).to(device) + self.upscale_net = ARCH_REGISTRY.get('CodeFormer')(dim_embd=512, codebook_size=1024, n_head=8, n_layers=9, + connect_list=['32', '64', '128',"256"]).to(device) ckpt_path = load_file_from_url(url=pretrain_model_url, model_dir=ckpt_dir, progress=True, file_name=None) checkpoint = torch.load(ckpt_path)['params_ema'] diff --git a/web/js/uploadVideo.js b/web/js/uploadVideo.js index 869547d..976a49e 100644 --- a/web/js/uploadVideo.js +++ b/web/js/uploadVideo.js @@ -195,7 +195,7 @@ ComfyWidgets.VIDEOPLOAD = videoUpload; app.registerExtension({ name: "DHLive.UploadVideo", async beforeRegisterNodeDef(nodeType, nodeData, app) { - if (nodeData?.name == "LoadVideo") { + if (nodeData?.name == "DHLIVELoadVideo") { nodeData.input.required.upload = ["VIDEOPLOAD"]; } },