from transformers import LlamaForCausalLM, CodeLlamaTokenizer
model = LlamaForCausalLM.from_pretrained("codellama/CodeLlama-7b-hf")