python - BERTopic model.find_topics(text) throwing error AttributeError: 'Transformer' object has no attribute 'do_lower_case'

Question

以前我制作了一个模型并将其保存在我的设备上。现在，当我尝试加载模型并获取单词的主题时，我收到了这个错误。

这是我的代码：

model = BERTopic.load('/content/drive/MyDrive/Colab Notebooks/model')
pred = model.find_topics("course")[0][0]

这是我得到的错误：

Exception ignored in: <generator object tqdm_notebook.__iter__ at 0x7fa8b86c9650>
Traceback (most recent call last):
  File "/usr/local/lib/python3.7/dist-packages/tqdm/notebook.py", line 220, in __iter__
    self.sp(bar_style='danger')
AttributeError: 'tqdm_notebook' object has no attribute 'sp'
---------------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-70-cf00a7f54361> in <module>()
      1   model = BERTopic.load('/content/drive/MyDrive/Colab Notebooks/model')
----> 2   pred = model.find_topics("course")[0][0]

5 frames
/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py in __getattr__(self, name)
    946                 return modules[name]
    947         raise AttributeError("'{}' object has no attribute '{}'".format(
--> 948             type(self).__name__, name))
    949 
    950     def __setattr__(self, name: str, value: Union[Tensor, 'Module']) -> None:

AttributeError: 'Transformer' object has no attribute 'do_lower_case'

我无法弄清楚，非常感谢任何帮助。

标签： python