Magic-AI

/

Starlight-1

like 0

Follow

Magic AI 4

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="Magic-AI/Starlight-1")

# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM

tokenizer = AutoTokenizer.from_pretrained("Magic-AI/Starlight-1")
model = AutoModelForMaskedLM.from_pretrained("Magic-AI/Starlight-1")