Ddroidlabs-Codex-mini / load-pretrained-model.py
dnnsdunca's picture
Upload 19 files
1f6a5b9 verified
raw
history blame
No virus
301 Bytes
# load_pretrained_model.py
from transformers import AutoModelForCausalLM, AutoTokenizer
def load_model_and_tokenizer():
model_name = "gpt-3"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
return model, tokenizer