How to use ydshieh/tiny-random-BertForNextSentencePrediction with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForNextSentencePrediction tokenizer = AutoTokenizer.from_pretrained("ydshieh/tiny-random-BertForNextSentencePrediction") model = AutoModelForNextSentencePrediction.from_pretrained("ydshieh/tiny-random-BertForNextSentencePrediction", device_map="auto")
Upload tiny models for BertForNextSentencePrediction
· Sign up or log in to comment
