gpt2_large / README.md
IMvision12's picture
Upload folder using huggingface_hub
32ca75a verified
|
Raw
History Blame Contribute Delete
767 Bytes
metadata
pipeline_tag: text-generation
license: mit
base_model: openai-community/gpt2-large
library_name: kerasformers
tags:
  - keras
  - kerasformers
  - text-generation
  - gpt2
  - pytorch
  - jax
  - tf

gpt2_large

Pure-Keras 3 GPT2 weights for kerasformers, converted from openai-community/gpt2-large.

Usage

from kerasformers.models.gpt2 import GPT2Generate, GPT2Tokenizer

model = GPT2Generate.from_weights("kerasformers/gpt2_large")
tokenizer = GPT2Tokenizer.from_weights("kerasformers/gpt2_large")
ids = tokenizer("The meaning of life is")
print(model.generate(ids, max_length=40))

License: mit, inherited from the upstream source.