ModuleNotFoundError: No module named 'agatha.ml.util.embedding_index'
Opened this issue · 3 comments
p123hx commented
Hi,
When following the README to loading the Agatha model in
model = torch.load("<AGATHA_DATA_DIR>/hypothesis_predictor_512/model.pt")
, it prompts
ModuleNotFoundError: No module named 'agatha.ml.util.embedding_index'.
I looked up the directory agatha/ml/util/ and there is no file named embedding_index.
Could u tell how to solve it?
Thx
p123hx commented
It seems that in commit 8b4cd22 embedding_index is replaced by embedding_lookup.
Need to replace the file name accordingly.
HarrySong123 commented
It seems that the pretrained model and the latest codes do not match. Replacing the file name of embedding_lookup is not enough since the repo also lacks some scripts that is asked by the pretrained model.
amir-hassan25 commented
I am facing the same issue.