search

jxhe / efficient-knnlm

Pytorch implementation of paper "Efficient Nearest Neighbor Language Models" (EMNLP 2021)
MIT License
71 stars 6 forks source link

where is "datasets" at "from datasets import load_dataset" of "/ef_knnlm/adaptive_retrieval/adaptive_retrieval.py" #1

Closed Huisw closed 2 years ago

Huisw commented 2 years ago

🐛 Bug

bash ef_knnlm/adaptive_retrieval/train_ar.sh

Traceback (most recent call last): File "ef_knnlm/adaptive_retrieval/adaptive_retrieval.py", line 13, in from datasets import load_dataset ImportError: cannot import name 'load_dataset' from 'datasets' (unknown location)

how to get "datasets"?

jxhe commented 2 years ago

I think you can just run pip install datasets