huggingface-transformers - Rasa 与 HuggingFace 的集成管道

Question

我正在尝试将 Rasa 配置为使用在 HuggingFace 预训练的模型。这是我的设置

language: fr
pipeline:
    - name: LanguageModelTokenizer
    - name: LanguageModelFeaturizer
      model_name: "bert"
      model_weights: "setu4993/LaBSE"
      cache_dir: "/rasa/model"

HuggingFace 模型在这里

在我运行“rasa train”后，它显示 OSError: Model name “setu4993/LaBSE” is not found。知道如何将这个拥抱脸模型集成到 Rasa 中吗？

我们可以在 Rasa 配置中包含多个 HuggingFace 管道吗？例子

pipeline:
    - model_name: "bert"
...
    - model_name: "xlnet"

标签： huggingface-transformersrasa

Answer 1

为了在config.yml文件中的管道中使用 HuggingFace Transformers，您必须执行以下操作：

1. 在您的终端中运行以下命令pip install rasa[transformers] --use-feature=2020-solver
2. 在您的config.yml文件中添加以下内容：

   - name: HFTransformersNLP
     model_name: "bert" # Name of the language model to use
     model_weights: "rasa/LaBSE" # Pre-Trained weights to be loaded
   - name: LanguageModelTokenizer
   - name: LanguageModelFeaturizer