Statcan Dialogue Dataset & Models
Collection
mcgill-nlp.github.io/statcan-dialogue-dataset • 18 items • Updated • 4
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("McGill-NLP/tapas-statcan-large-conversation_encoder-title")
model = AutoModel.from_pretrained("McGill-NLP/tapas-statcan-large-conversation_encoder-title", device_map="auto")Quick Links
No model card
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="McGill-NLP/tapas-statcan-large-conversation_encoder-title", device_map="auto")