CodeLLaMA-Linux-BugFix / train /download_model.py
Mac Huang
Push full project including dataset, code, and training scripts
ed6b901
raw
history blame
264 Bytes
from transformers import AutoTokenizer, AutoModelForCausalLM
model = AutoModelForCausalLM.from_pretrained("codellama/CodeLlama-7b-Instruct-hf")
tokenizer = AutoTokenizer.from_pretrained("codellama/CodeLlama-7b-Instruct-hf")
print("✅ Download complete.")