Rasa与HuggingFace的集成管道

Question

我正在尝试将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:找不到型号名称“setu4993/LaBSE”。你知道如何将这个huggingface模型集成到Rasa中吗？

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

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

Answer 1

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

在终端pip install rasa[transformers] --use-feature=2020-solver

• in config.yml文件中运行以下命令，然后添加以下内容：- name: HFTransformersNLP model_name："bert“#要使用model_weights的语言模型的名称："rasa/LaBSE”#要加载的预先训练的权重- name: LanguageModelTokenizer - name: rasa