mistralai-distributed / model_load.py
changgyu's picture
Upload 19 files
668bf5d verified
from transformers import AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.3")