Rasa 与 HuggingFace 的集成管道答案

【问题标题】：Rasa integration pipeline with HuggingFaceRasa 与 HuggingFace 的集成管道
【发布时间】：2021-07-03 21:14:29
【问题描述】：

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

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

The HuggingFace model is here

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

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

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

【问题讨论】：

标签： huggingface-transformers rasa

【解决方案1】：

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

在终端中运行以下命令pip install rasa[transformers] --use-feature=2020-solver

在您的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

【讨论】：