# Load model directly
from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained("TheDiamondKing/Discord-Message-Small")
model = AutoModelForMaskedLM.from_pretrained("TheDiamondKing/Discord-Message-Small")Quick Links
Simple model trained with 2790 Discord messages ( Might have some NSFW responses )
- Downloads last month
- 7
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="TheDiamondKing/Discord-Message-Small")