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：没有找到型号名称"setu4993/LaBSE"。知道如何将这个拥抱脸模型集成到Rasa中吗？

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

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

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