metadata
pipeline_tag: text-generation
license: mit
base_model: openai-community/gpt2-medium
library_name: kerasformers
tags:
- keras
- kerasformers
- text-generation
- gpt2
- pytorch
- jax
- tf
gpt2_medium
Pure-Keras 3 GPT2 weights for kerasformers, converted from openai-community/gpt2-medium.
Usage
from kerasformers.models.gpt2 import GPT2Generate, GPT2Tokenizer
model = GPT2Generate.from_weights("kerasformers/gpt2_medium")
tokenizer = GPT2Tokenizer.from_weights("kerasformers/gpt2_medium")
ids = tokenizer("The meaning of life is")
print(model.generate(ids, max_length=40))
License: mit, inherited from the upstream source.