stanford-futuredata / Baleen

Baleen: Robust Multi-Hop Reasoning at Scale via Condensed Retrieval (NeurIPS'21)
MIT License
42 stars 5 forks source link

RuntimeError when running hover_inference.py #4

Closed hyukkyukang closed 1 year ago

hyukkyukang commented 1 year ago

Hello,

I encountered a RuntimeError when executing hover_inference.py as instructed in the README. The error seems to originate from the ColBERT package:

image

This appears to be a known issue in the ColBERT package, which has been resolved in its main branch.

However, using the main branch of ColBERT leads to other errors, which appear to be caused by API mismatches:

image

Are there any plans to make the code compatible with the recent version of ColBERT?

hyukkyukang commented 1 year ago

The problem only happens with the recent version of Pytorch. Using the environment settings from ColBERT/conda_env.yml has resolved the issue.