# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Vaylenid/text-generation-demo", device_map="auto")Quick Links
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Vaylenid/text-generation-demo", device_map="auto")
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="Vaylenid/text-generation-demo")