Track the model files 7cb5a13
Amr Keleg commited on
How to use VARabi/ALDi-Token-DI with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="VARabi/ALDi-Token-DI") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification
tokenizer = AutoTokenizer.from_pretrained("VARabi/ALDi-Token-DI")
model = AutoModelForTokenClassification.from_pretrained("VARabi/ALDi-Token-DI", device_map="auto")