runtime error

Exit code: 1. Reason: rch_ie/auto.py", line 132, in from_pretrained taskmodule = AutoTaskModule.from_pretrained( File "/usr/local/lib/python3.10/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) File "/usr/local/lib/python3.10/site-packages/pytorch_ie/core/hf_hub_mixin.py", line 220, in from_pretrained return cls._from_pretrained( File "/usr/local/lib/python3.10/site-packages/pytorch_ie/auto.py", line 98, in _from_pretrained class_name = config.pop(cls.config_type_key) KeyError: 'taskmodule_type' The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/user/app/src/demo/annotation_utils.py", line 113, in load_argumentation_model model = instantiate(config, **kwargs) File "/usr/local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 226, in instantiate return instantiate_node( File "/usr/local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 347, in instantiate_node return _call_target(_target_, partial, args, kwargs, full_key) File "/usr/local/lib/python3.10/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 97, in _call_target raise InstantiationException(msg) from e hydra.errors.InstantiationException: Error in call to target 'src.pipeline.ner_re_pipeline.NerRePipeline': KeyError('taskmodule_type') During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/user/app/src/start_demo.py", line 114, in main argumentation_model = load_argumentation_model( File "/home/user/app/src/demo/annotation_utils.py", line 116, in load_argumentation_model raise gr.Error(f"Failed to load argumentation model: {e}") gradio.exceptions.Error: "Failed to load argumentation model: Error in call to target 'src.pipeline.ner_re_pipeline.NerRePipeline':\nKeyError('taskmodule_type')" Set the environment variable HYDRA_FULL_ERROR=1 for a complete stack trace.

Container logs: