search

pawansharmaaaa / Lip_Wise

Orchestrating AI for stunning lip-synced videos. Effortless workflow, exceptional results, all in one place.
Apache License 2.0
53 stars 14 forks source link

KeyError: "No object named 'CodeFormer' found in 'arch' registry!" #14

Open tranthuhoai3786 opened 2 months ago

tranthuhoai3786 commented 2 months ago

@pawansharmaaaa I run "python launch.py" and I have GUI at http://127.0.0.1:7860/. When I chose video and audio with the same length then press "process video", i get error:

Traceback (most recent call last): File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/gradio/queueing.py", line 527, in process_events response = await route_utils.call_process_api( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/gradio/route_utils.py", line 261, in call_process_api output = await app.get_blocks().process_api( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/gradio/blocks.py", line 1788, in process_api result = await self.call_function( ^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/gradio/blocks.py", line 1340, in call_function prediction = await anyio.to_thread.run_sync( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/anyio/to_thread.py", line 56, in run_sync return await get_async_backend().run_sync_in_worker_thread( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/anyio/_backends/_asyncio.py", line 2144, in run_sync_in_worker_thread return await future ^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/anyio/_backends/_asyncio.py", line 851, in run result = context.run(func, args) ^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/gradio/utils.py", line 759, in wrapper response = f(args, *kwargs) ^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^ File "/media/ai-3090/hdd_2/hoaitt/Lip_Wise/infer.py", line 308, in infer_video ml = model_loaders.ModelLoader(face_restorer, weight) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/media/ai-3090/hdd_2/hoaitt/Lip_Wise/helpers/model_loaders.py", line 32, in init self.restorer = self.load_codeformer_model() ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/media/ai-3090/hdd_2/hoaitt/Lip_Wise/helpers/model_loaders.py", line 151, in load_codeformer_model model = ARCH_REGISTRY.get('CodeFormer')(dim_embd=512, codebook_size=1024, n_head=8, n_layers=9, connect_list=['32', '64', '128', '256']).to(self.device) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ai-3090/anaconda3/envs/lipwise/lib/python3.11/site-packages/basicsr/utils/registry.py", line 71, in get raise KeyError(f"No object named '{name}' found in '{self._name}' registry!") KeyError: "No object named 'CodeFormer' found in 'arch' registry!"

pawansharmaaaa commented 2 months ago

Please run the setup file first.

Zachysaurs commented 1 month ago

same error

pawansharmaaaa commented 1 month ago

This error arises when the archs directory is not copied to the codedormer directory, you can manually paste the archs folder at {Venv Directory}/bin/codedormer/ to resolve this issue. This is handled when you run setup file but there are some edge cases where it doesn't work.