codellama

/

CodeLlama-7b-hf

like 329

Follow

Code Llama 511

Get Error

OSError: Unable to load weights from pytorch checkpoint file for '/mnt/bd/fazzie-data/models/CodeLlama-7b-hf/pytorch_model-00001-of-00002.bin' at '/mnt/bd/fazzie-data/models/CodeLlama-7b-hf/pytorch_model-00001-of-00002.bin'. If you tried to load a PyTorch model from a TF 2.0 checkpoint, please set from_tf=True.

Here is the code

from transformers import AutoTokenizer, AutoModelForCausalLM
import transformers
import torch

model = "./models/CodeLlama-7b-hf"

tokenizer = AutoTokenizer.from_pretrained(model)

model = AutoModelForCausalLM.from_pretrained(model)