加载微调模型报错

[unira-zwj]

加载微调模型的部分为：

    model, model_args = AutoModel.from_pretrained(
        ‘/data/xxx/workspace/VisualGLM-6B-main/checkpoints/finetune-visualglm-6b-07-10-10-16’,
        args=argparse.Namespace(
        fp16=True,
        skip_init=True,
        use_gpu_initialization=True if (torch.cuda.is_available() and args.quant is None) else False,
        device='cuda' if (torch.cuda.is_available() and args.quant is None) else 'cpu',
    ))

报错信息为： ValueError: model_class FineTuneVisualGLMModel not found.

请问为什么加载微调模型会报找不到这个模型类别的错误

[Sleepychord]

AutoModel 换为新定义的FineTuneVisualGLMModel即可

[unira-zwj]

AutoModel 换为新定义的FineTuneVisualGLMModel即可

感谢，已经改好了，使用了finetune.py里面的FineTuneVisualGLMModel