Using GPT as shown in the paper about language-models-are-knowledge-graphs-pytorch HOT 4 OPEN

Oh I found it here. Turns out the choices are:

'bert-large-uncased', 'bert-large-cased', 'bert-base-uncased', 'bert-base-cased', 'gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-xl'

from language-models-are-knowledge-graphs-pytorch.

I attempted to run it with the gpt2-xl parameters, however I got the following error:
Traceback (most recent call last):
File "/usr/lib/python3.7/multiprocessing/pool.py", line 121, in worker
result = (True, func(*args, **kwds))
File "/content/language-models-are-knowledge-graphs-pytorch/process.py", line 29, in bfs
return BFS(s, end, graph, max_size, black_list_relation)
File "/content/language-models-are-knowledge-graphs-pytorch/utils.py", line 19, in BFS
visited = [False] * (max(graph.keys())+100)
ValueError: max() arg is an empty sequence
"""

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
File "extract.py", line 74, in
for triplets in parse_sentence(sent.text, tokenizer, encoder, nlp, use_cuda=use_cuda):
File "/content/language-models-are-knowledge-graphs-pytorch/process.py", line 94, in parse_sentence
for output in pool.imap_unordered(bfs, params):
File "/usr/lib/python3.7/multiprocessing/pool.py", line 748, in next
raise value
File "/usr/lib/python3.7/multiprocessing/pool.py", line 121, in worker
result = (True, func(*args, **kwds))
File "/content/language-models-are-knowledge-graphs-pytorch/process.py", line 29, in bfs
return BFS(s, end, graph, max_size, black_list_relation)
File "/content/language-models-are-knowledge-graphs-pytorch/utils.py", line 19, in BFS
visited = [False] * (max(graph.keys())+100)
ValueError: max() arg is an empty sequence

from language-models-are-knowledge-graphs-pytorch.

Also tried running it with just gpt2-large, same error and empty output file

from language-models-are-knowledge-graphs-pytorch.

got this error too

from language-models-are-knowledge-graphs-pytorch.