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

[tranthuhoai3786]

@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]

Please run the setup file first.

[Zachysaurs]

same error

[pawansharmaaaa]

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.