Open sunxiaoyu12 opened 1 year ago
把这个仓库的tokenizer相关的文件下载到本地(不需要下载模型权重文件):https://huggingface.co/THUDM/chatglm-6b
然后把这行代码换成下面的代码:https://github.com/THUDM/VisualGLM-6B/blob/main/finetune_visualglm.py#L157
from transformers import AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained('your/path/', trust_remote_code=True)
并删掉这行代码:https://github.com/THUDM/VisualGLM-6B/blob/main/finetune_visualglm.py#L181
把这个仓库的tokenizer相关的文件下载到本地(不需要下载模型权重文件):https://huggingface.co/THUDM/chatglm-6b
然后把这行代码换成下面的代码:https://github.com/THUDM/VisualGLM-6B/blob/main/finetune_visualglm.py#L157
from transformers import AutoTokenizer tokenizer = AutoTokenizer.from_pretrained('your/path/', trust_remote_code=True)
并删掉这行代码:https://github.com/THUDM/VisualGLM-6B/blob/main/finetune_visualglm.py#L181
多谢!可以啦!
微调时报错如下:
应该是调用sat库的get_tokenizer出错了,有没有办法可以调用本地的tokenizer?