ITADN

hydra.errors.InstantiationException: Error in call to target 'relik.retriever.pytorch_modules.model.GoldenRetriever'

#33Closedchocwaffles 创建于 2025-03-24
C
chocwafflescommented
Hi, Can't seem to get this code started. please help. hydra.errors.InstantiationException: Error in call to target 'relik.retriever.pytorch_modules.model.GoldenRetriever': ConnectionError(ProtocolError('Connection aborted.', ConnectionResetError(104, 'Connection reset by peer')) Installed without errors. ``` pip install relik ``` ran this code ``` from relik import Relik from relik.inference.data.objects import RelikOutput relik = Relik.from_pretrained("relik-ie/relik-relation-extraction-small") relik_out: RelikOutput = relik("Michael Jordan was one of the best players in the NBA.") ``` full trace error ``` /home/ubuntu/repos/rem/.venv/bin/python /home/ubuntu/repos/rem/src/test_relik.py ___ __ /\_ \ __ /\ \ _ __ __ \//\ \ /\_\ \ \ \/'\ /\`'__\ /'__`\ \ \ \ \/\ \ \ \ , < \ \ \/ /\ __/ \_\ \_ \ \ \ \ \ \\`\ \ \_\ \ \____\ /\____\ \ \_\ \ \_\ \_\ \/_/ \/____/ \/____/ \/_/ \/_/\/_/ /home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py:797: FutureWarning: `resume_download` is deprecated and will be removed in version 1.0.0. Downloads always resume when possible. If you want to force a new download, use `force_download=True`. warnings.warn( [2025-03-24 22:54:43,290] [INFO] [relik.inference.annotator.from_pretrained:700] [PID:10258] [RANK:0] Loading Relik from relik-ie/relik-relation-extraction-small [2025-03-24 22:54:43,292] [INFO] [relik.inference.annotator.from_pretrained:701] [PID:10258] [RANK:0] { '_target_': 'relik.inference.annotator.Relik', 'index': { 'triplet': { '_target_': 'relik.retriever.indexers.inmemory.InMemoryDocumentIndex.from_pretrained', 'name_or_path': 'relik-ie/encoder-e5-small-v2-wikipedia-relations-index', }, }, 'metadata_fields': [], 'reader': { '_target_': 'relik.reader.pytorch_modules.triplet.RelikReaderForTripletExtraction', 'transformer_model': 'relik-ie/relik-reader-deberta-v3-small-re-wikipedia', }, 'retriever': { 'triplet': { '_target_': 'relik.retriever.pytorch_modules.model.GoldenRetriever', 'question_encoder': 'relik-ie/encoder-e5-small-v2-wikipedia-relations', }, }, 'task': 'TRIPLET', 'top_k': 30, 'window_size': 64, 'window_stride': 32, } Traceback (most recent call last): File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 787, in urlopen response = self._make_request( ^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 488, in _make_request raise new_e File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 464, in _make_request self._validate_conn(conn) File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 1093, in _validate_conn conn.connect() File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connection.py", line 741, in connect sock_and_verified = _ssl_wrap_socket_and_match_hostname( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connection.py", line 920, in _ssl_wrap_socket_and_match_hostname ssl_sock = ssl_wrap_socket( ^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/ssl_.py", line 460, in ssl_wrap_socket ssl_sock = _ssl_wrap_socket_impl(sock, context, tls_in_tls, server_hostname) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/ssl_.py", line 504, in _ssl_wrap_socket_impl return ssl_context.wrap_socket(sock, server_hostname=server_hostname) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.11/ssl.py", line 517, in wrap_socket return self.sslsocket_class._create( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.11/ssl.py", line 1075, in _create self.do_handshake() File "/usr/local/lib/python3.11/ssl.py", line 1346, in do_handshake self._sslobj.do_handshake() ConnectionResetError: [Errno 104] Connection reset by peer During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/requests/adapters.py", line 667, in send resp = conn.urlopen( ^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 841, in urlopen retries = retries.increment( ^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/retry.py", line 474, in increment raise reraise(type(error), error, _stacktrace) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/util.py", line 38, in reraise raise value.with_traceback(tb) File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 787, in urlopen response = self._make_request( ^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 488, in _make_request raise new_e File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 464, in _make_request self._validate_conn(conn) File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connectionpool.py", line 1093, in _validate_conn conn.connect() File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connection.py", line 741, in connect sock_and_verified = _ssl_wrap_socket_and_match_hostname( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/connection.py", line 920, in _ssl_wrap_socket_and_match_hostname ssl_sock = ssl_wrap_socket( ^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/ssl_.py", line 460, in ssl_wrap_socket ssl_sock = _ssl_wrap_socket_impl(sock, context, tls_in_tls, server_hostname) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/urllib3/util/ssl_.py", line 504, in _ssl_wrap_socket_impl return ssl_context.wrap_socket(sock, server_hostname=server_hostname) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.11/ssl.py", line 517, in wrap_socket return self.sslsocket_class._create( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/usr/local/lib/python3.11/ssl.py", line 1075, in _create self.do_handshake() File "/usr/local/lib/python3.11/ssl.py", line 1346, in do_handshake self._sslobj.do_handshake() urllib3.exceptions.ProtocolError: ('Connection aborted.', ConnectionResetError(104, 'Connection reset by peer')) During handling of the above exception, another exception occurred: Traceback (most recent call last): File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 92, in _call_target return _target_(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/relik/retriever/pytorch_modules/model.py", line 65, in __init__ question_encoder = GoldenRetrieverModel.from_pretrained( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/transformers/modeling_utils.py", line 3351, in from_pretrained resolved_archive_file = cached_file(pretrained_model_name_or_path, filename, **cached_file_kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/transformers/utils/hub.py", line 399, in cached_file resolved_file = hf_hub_download( ^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/utils/_validators.py", line 114, in _inner_fn return fn(*args, **kwargs) ^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 862, in hf_hub_download return _hf_hub_download_to_cache_dir( ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1011, in _hf_hub_download_to_cache_dir _download_to_tmp_and_move( File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 1547, in _download_to_tmp_and_move http_get( File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 371, in http_get r = _request_wrapper( ^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/file_download.py", line 303, in _request_wrapper response = get_session().request(method=method, url=url, **params) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/requests/sessions.py", line 589, in request resp = self.send(prep, **send_kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/requests/sessions.py", line 703, in send r = adapter.send(request, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/huggingface_hub/utils/_http.py", line 96, in send return super().send(request, *args, **kwargs) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/requests/adapters.py", line 682, in send raise ConnectionError(err, request=request) requests.exceptions.ConnectionError: (ProtocolError('Connection aborted.', ConnectionResetError(104, 'Connection reset by peer')), '(Request ID: fb1cceec-dc0f-45f0-b6f8-27ecdb594fb8)') The above exception was the direct cause of the following exception: Traceback (most recent call last): File "/home/ubuntu/repos/rem/src/test_relik.py", line 6, in <module> relik = Relik.from_pretrained("relik-ie/relik-relation-extraction-small") ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/relik/inference/annotator.py", line 724, in from_pretrained retriever = load_retriever( ^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/relik/inference/utils.py", line 150, in load_retriever _retriever[TaskType.TRIPLET] = _instantiate_retriever( ^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/relik/inference/utils.py", line 42, in _instantiate_retriever retriever = hydra.utils.instantiate( ^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 226, in instantiate return instantiate_node( ^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/site-packages/hydra/_internal/instantiate/_instantiate2.py", line 347, in instantiate_node return _call_target(_target_, partial, args, kwargs, full_key) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/home/ubuntu/repos/rem/.venv/lib/python3.11/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 'relik.retriever.pytorch_modules.model.GoldenRetriever': ConnectionError(ProtocolError('Connection aborted.', ConnectionResetError(104, 'Connection reset by peer')), '(Request ID: xxx') ```
关闭于 2025-03-25 0 条评论