metadata
license: mit
How to use
Requirements:
git clone https://github.com/ddevaul/desformers
cd desformers
pip install -r requirements.txt
cd ..
Now open your file and add this:
import sys
import torch
sys.path.append('./desformers/src')
from torch.utils.checkpoint import checkpoint
from transformers2 import BertConfig, BertTokenizer
from transformers2.models.bert import BertForMaskedLM
preload_path = 'cabrooks/character-level-logion'
char_tokenizer = BertTokenizer.from_pretrained(preload_path)
wordpiece_tokenizer = BertTokenizer.from_pretrained("cabrooks/LOGION-50k_wordpiece")
config = BertConfig()
config.word_piece_vocab_size = 50000
config.vocab_size = char_tokenizer.vocab_size
config.char_tokenizer = char_tokenizer
config.wordpiece_tokenizer = wordpiece_tokenizer
config.max_position_embeddings = 1024
config.device2 = device
model = BertForMaskedLM(config).to(device)
Download the weights from "my_custom_model.pth". Load these weights into the model:
model.load_state_dict(torch.load('my_custom_model.pth', map_location=torch.device('cpu')))
You are now ready to use the model.
Author:
This model was developed by Desmond DeVaul for his senior thesis at Princeton University.
It was built on the work of the Logion team at Princeton: https://www.logionproject.princeton.edu.