Trained model 5cb2345
ravil99 commited on
How to use 4ku/gpt2-persona-sponge_bob with Transformers:
# Load model directly
from transformers import AutoTokenizer, AutoModel
tokenizer = AutoTokenizer.from_pretrained("4ku/gpt2-persona-sponge_bob")
model = AutoModel.from_pretrained("4ku/gpt2-persona-sponge_bob", device_map="auto")