OSError: 找不到名为['pytorch_model.bin'、'tf_model.h5'、'model.ckpt.index']的文件。

Question

当我在线加载BERT预训练模型时，会出现此错误：OSError: Error no file named['pytorch_model.bin'，'tf_model.h5'，'model.ckpt.index'] found in directory uncased_L-12_H-768_A-12 or 'from_tf' set to False 我该怎么办？

- Asma

---

对我来说解决问题的方法是升级到最新版本的transformers库。 - undefined

Answer 1

我找到了以下内容。请前往下面的链接，点击圆圈以下载文件，将其重命名为 pytorch_model.bin 并放置在 biobert-nli 目录中，这样问题就解决了。不知道如何从链接克隆。

https://huggingface.co/gsarti/biobert-nli/tree/main

- caot

Answer 2

这些是预训练模型文件。如果提供的话，你可以从你拿到代码的存储库中下载它们。如果你使用的是HuggingFace，你可以在他们的网站上获取这些文件。

- Abhishek Verma

Answer 3

你可以使用transformers-cli（pip install transformers）来转换bert。

transformers-cli convert --model_type bert \
  --tf_checkpoint $BERT_BASE_DIR/model.ckpt \
  --config $BERT_BASE_DIR/config.json \
  --pytorch_dump_output $BERT_BASE_DIR/pytorch_model.bin

- akili

---

需要已经安装了“datasets”才能提取“pytorch_model.bin”，如果没有，只需通过pip“pip install datasets”即可。 - Vahab