transformers/temp_llama.py

from transformers import LlamaForCausalLM, CodeLlamaTokenizer

model = LlamaForCausalLM.from_pretrained("codellama/CodeLlama-7b-hf")
model = LlamaForCausalLM.from_pretrained("codellama/CodeLlama-7b-hf")