Statcan Dialogue Dataset & Models
Collection
mcgill-nlp.github.io/statcan-dialogue-dataset โข 18 items โข Updated โข 4
How to use McGill-NLP/dpr-statcan-metadata_encoder-basic_info with Transformers:
# pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, DPRContextEncoder
tokenizer = AutoTokenizer.from_pretrained("McGill-NLP/dpr-statcan-metadata_encoder-basic_info")
model = DPRContextEncoder.from_pretrained("McGill-NLP/dpr-statcan-metadata_encoder-basic_info", device_map="auto")No model card