首页 > 解决方案 > DISTILBERT_BASE_UNCASED 加载失败 - 拥抱脸变压器

问题描述

我正在尝试执行:

import ktrain

from ktrain import text

MODEL_NAME='distilbert-base-uncased'

t=text.Transformer(MODEL_NAME, maxlen=500, classes=np.unique(y_train))

我收到以下错误:

*OSError: Model name 'distilbert-base-uncased' was not found in tokenizers model name list (distilbert-base-uncased, distilbert-base-uncased-distilled-squad, distilbert-base-cased, distilbert-base-cased-distilled-squad, distilbert-base-german-cased, distilbert-base-multilingual-cased). We assumed 'distilbert-base-uncased' was a path, a model identifier, or url to a directory containing vocabulary files named ['vocab.txt'] but couldn't find such vocabulary files at this path or url.*

ktrain 版本 0.14.6 变压器版本 2.8.0

库是使用 pip install 安装的。

任何帮助,将不胜感激。

标签: huggingface-transformersbert-language-model

解决方案


推荐阅读