VisualGLM-6B
VisualGLM-6B copied to clipboard
加载微调后模型出错
model, args = FineTuneVisualGLMModel.from_pretrained('/mnt/dolphinfs/hdd_pool/docker/user/hadoop-search/ganshu01/projects/VisualGLM-6B-main/output/finetune-visualglm-6b-06-20-14-50', args=argparse.Namespace( fp16=True, skip_init=True, use_gpu_initialization=False, device='cpu', ))
ValueError: Missing keys for inference: ['transformer.layers.0.attention.dense.matrix_A',
'transformer.layers.0.attention.dense.matrix_B', 'transformer.layers.14.attention.dense.matrix_A', 'transformer.layers.14.attention.dense.matrix_B']. If you still want to inference anyway, pass --force_inference to args.