Neuracraft / model_loading.py
Erfan11's picture
Rename Openai.ai to model_loading.py
8117905 verified
raw
history blame
361 Bytes
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-generation", model="openai-community/gpt2")
# Load model directly
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("openai-community/gpt2")
model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2")