init repo 6cc29c9
Carlos E. Jimenez commited on
How to use closji/wiki103_bert_small_k10_e27 with Transformers:
# Load model directly
from transformers import AutoTokenizer, CoLwithBert
tokenizer = AutoTokenizer.from_pretrained("closji/wiki103_bert_small_k10_e27")
model = CoLwithBert.from_pretrained("closji/wiki103_bert_small_k10_e27", device_map="auto")