怎么下载模型呢？
#1
by
Yutong001
- opened
我的代码是
from transformers import MistralForCausalLM
model = MistralForCausalLM.from_pretrained(model_name )
提示OSError: Error no file named pytorch_model.bin, model.safetensors, tf_model.h5, model.ckpt.index or flax_model.msgpack found in directory OpenRLHF/Mistral-7b-PRM-Math-Shepherd.
请问怎么把这个模型下载下来呢？
直接 AutoModelForCausalLM.from_pretrained